From e00e5f308aeb364e50fbc1e0365bac302683afbd Mon Sep 17 00:00:00 2001 From: Gaspar Rochette Date: Fri, 13 Feb 2026 15:54:26 +0000 Subject: [PATCH 1/2] fix: change moe model for compatibility with earlier transformers versions --- tests/fixtures.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/fixtures.py b/tests/fixtures.py index 1c2f24c1..7ff8d4e1 100644 --- a/tests/fixtures.py +++ b/tests/fixtures.py @@ -198,6 +198,6 @@ def get_autoregressive_text_to_image_model(model_id: str) -> tuple[Any, SmashCon "flux_tiny": partial(get_diffusers_model, "pruna-test/tiny_flux", torch_dtype=torch.float16), "tiny_llama": partial(get_automodel_transformers, "pruna-test/tiny_llama", torch_dtype=torch.bfloat16), "qwen3_next_moe_tiny_random": partial( - get_automodel_transformers, "tiny-random/qwen3-next-moe", torch_dtype=torch.bfloat16 + get_automodel_transformers, "yujiepan/qwen1.5-moe-tiny-random", torch_dtype=torch.bfloat16 ), } From 97b3c39831a75e364928c830eade2418b6e8d0f4 Mon Sep 17 00:00:00 2001 From: Gaspar Rochette Date: Fri, 13 Feb 2026 15:56:35 +0000 Subject: [PATCH 2/2] fix: rename fixture to fit ealier model --- tests/algorithms/testers/reduce_noe.py | 2 +- tests/fixtures.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/algorithms/testers/reduce_noe.py b/tests/algorithms/testers/reduce_noe.py index 84344123..9b8df4c3 100644 --- a/tests/algorithms/testers/reduce_noe.py +++ b/tests/algorithms/testers/reduce_noe.py @@ -6,7 +6,7 @@ class TestReduceNOE(AlgorithmTesterBase): """Test the ReduceNOE algorithm.""" - models = ["qwen3_next_moe_tiny_random"] + models = ["qwen_moe_tiny_random"] reject_models = ["sd_tiny_random"] allow_pickle_files = False algorithm_class = ReduceNOE diff --git a/tests/fixtures.py b/tests/fixtures.py index 7ff8d4e1..f9bb2f7f 100644 --- a/tests/fixtures.py +++ b/tests/fixtures.py @@ -197,7 +197,7 @@ def get_autoregressive_text_to_image_model(model_id: str) -> tuple[Any, SmashCon "wan_tiny_random": partial(get_diffusers_model, "pruna-test/wan-t2v-tiny-random", torch_dtype=torch.bfloat16), "flux_tiny": partial(get_diffusers_model, "pruna-test/tiny_flux", torch_dtype=torch.float16), "tiny_llama": partial(get_automodel_transformers, "pruna-test/tiny_llama", torch_dtype=torch.bfloat16), - "qwen3_next_moe_tiny_random": partial( + "qwen_moe_tiny_random": partial( get_automodel_transformers, "yujiepan/qwen1.5-moe-tiny-random", torch_dtype=torch.bfloat16 ), }