tests: add opt fixture for distillation config checks. (#570)

begumcig · web-flow · commit ab8dccfb4099 · 2026-03-09T09:31:51.000+01:00
diff --git a/tests/fixtures.py b/tests/fixtures.py
@@ -212,4 +212,5 @@ def get_autoregressive_text_to_image_model(model_id: str) -> tuple[Any, SmashCon
     "qwen_moe_tiny_random": partial(
         get_automodel_transformers, "yujiepan/qwen1.5-moe-tiny-random", torch_dtype=torch.bfloat16
     ),
+    "opt_125m": partial(get_automodel_transformers, "facebook/opt-125m", torch_dtype=torch.bfloat16),
 }

Original file line number	Diff line number	Diff line change
`@@ -212,4 +212,5 @@ def get_autoregressive_text_to_image_model(model_id: str) -> tuple[Any, SmashCon`
`212`	`212`	`"qwen_moe_tiny_random": partial(`
`213`	`213`	`get_automodel_transformers, "yujiepan/qwen1.5-moe-tiny-random", torch_dtype=torch.bfloat16`
`214`	`214`	`),`
	`215`	`+ "opt_125m": partial(get_automodel_transformers, "facebook/opt-125m", torch_dtype=torch.bfloat16),`
`215`	`216`	`}`