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 1c2f24c1..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( - get_automodel_transformers, "tiny-random/qwen3-next-moe", torch_dtype=torch.bfloat16 + "qwen_moe_tiny_random": partial( + get_automodel_transformers, "yujiepan/qwen1.5-moe-tiny-random", torch_dtype=torch.bfloat16 ), }