Enable fx tracing for Mistral (#30209)

* tracing for mistral * typo * fix copies
2024-04-17 14:38:48 +05:00
parent 98717cb341
commit 304c6a1e0d
7 changed files with 9 additions and 6 deletions
--- a/tests/models/mixtral/test_modeling_mixtral.py
+++ b/tests/models/mixtral/test_modeling_mixtral.py
@@ -302,6 +302,7 @@ class MixtralModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
    )
    test_headmasking = False
    test_pruning = False
+    fx_compatible = True

    # TODO (ydshieh): Check this. See https://app.circleci.com/pipelines/github/huggingface/transformers/79245/workflows/9490ef58-79c2-410d-8f51-e3495156cf9c/jobs/1012146
    def is_pipeline_test_to_skip(