Update tests/openvino/test_diffusion.py

huggingface · Jan 30, 2025 · d8585db · d8585db
1 parent d293c56
commit d8585db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/openvino/test_diffusion.py b/tests/openvino/test_diffusion.py
@@ -110,7 +110,7 @@ def test_ov_pipeline_class_dispatch(self, model_arch: str):
 
         self.assertEqual(ov_pipeline.auto_model_class, auto_pipeline.__class__)
 
-        auto_pipeline = auto_pipeline.from_pretrained(MODEL_NAMES[model_arch])
+        auto_pipeline = DiffusionPipeline.from_pretrained(MODEL_NAMES[model_arch])
         ov_pipeline = OVDiffusionPipeline.from_pretrained(MODEL_NAMES[model_arch])
 
         self.assertEqual(ov_pipeline.auto_model_class, auto_pipeline.__class__)