extend rewrite list

Author: xadupre

onnx_diagnostic/torch_export_patches/patch_module_helper.py

@@ -60,10 +60,39 @@ def code_needing_rewriting(cls_name: str) -> Optional[List[Any]]:
         pprint.pprint(code_needing_rewriting("BartForConditionalGeneration"))
     """
     if cls_name in {
+        "AutoformerEncoderLayer",
+        "AutoformerEncoder",
+        "AutoformerForPrediction",
         "BartEncoderLayer",
+        "AutoformerModel",
         "BartForConditionalGeneration",
+        "BigBirdPegasusEncoderLayer",
+        "BigBirdPegasusForConditionalGeneration",
+        "BigBirdPegasusForQuestionAnswering",
+        "BigBirdPegasusForCausalLM",
+        "BlenderbotSmallEncoderLayer",
+        "BlenderbotSmallForConditionalGeneration",
+        "BlenderbotSmallForCausalLM",
+        "InformerEncoderLayer",
+        "InformerForPrediction",
+        "LEDEncoderLayer",
+        "LEDClassificationHead",
+        "LEDForConditionalGeneration",
+        "MarianEncoderLayer",
+        "MarianEncoder",
+        "MarianModel",
+        "MvpEncoderLayer",
+        "MvpPrompt",
+        "MvpForConditionalGeneration",
+        "MvpForSequenceClassification",
+        "MvpForQuestionAnswering",
+        "MvpForCausalLM",
+        "NllbMoeEncoderLayer",
+        "NllbMoeForConditionalGeneration",
         "PLBartEncoderLayer",
         "PLBartForConditionalGeneration",
+        "TimeSeriesTransformerEncoderLayer",
+        "TimeSeriesTransformerForPrediction",
     }:
         return _rewrite_bart_encoder_layer()
     return None