FX symbolic_trace: do not test decoder_inputs_embeds (huggingface#31840)

only test input_embeds, not decoder_input_embeds
haikuoxin · Jul 9, 2024 · 0abf5e8 · 0abf5e8
1 parent 952dfd4
commit 0abf5e8
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(