only test input_embeds, not decoder_input_embeds

fxmarty · Jul 8, 2024 · fc56efc · fc56efc
1 parent ae9dd02
commit fc56efc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -1215,7 +1215,7 @@ def _create_and_check_torch_fx_tracing(self, config, inputs_dict, output_loss=Fa
                             (past_mask, inputs_to_test[1]["attention_mask"]), dim=1
                         )
 
-            if "inputs_embeds" in inspect.signature(model.forward).parameters:
+            if "inputs_embeds" in inspect.signature(model.forward).parameters and not model.config.is_encoder_decoder:
                 inputs_to_test.append(
                     {
                         "inputs_embeds": torch.rand(