Skip to content

GitLab

Unverified Commit 0abf5e8e authored by fxmarty's avatar fxmarty Committed by GitHub
Browse files

FX symbolic_trace: do not test decoder_inputs_embeds (#31840) 

only test input_embeds, not decoder_input_embeds
parent 952dfd48
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_common.py
View file @ 0abf5e8e
......@@ -1215,7 +1215,7 @@ class ModelTesterMixin:
(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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment