Skip to content

Commit 4a94677

Browse files
committed
add test
1 parent 95e3c82 commit 4a94677

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/openvino/test_diffusion.py

+5
Original file line numberDiff line numberDiff line change
@@ -359,6 +359,11 @@ def test_load_and_save_pipeline_with_safety_checker(self):
359359
self.assertTrue(model_lib in ["diffusers", "transformers"])
360360
self.assertFalse(model_class.startswith("OV"))
361361
loaded_pipeline = self.OVMODEL_CLASS.from_pretrained(tmpdirname)
362+
for component in ["text_encoder", "unet", "vae_encoder", "vae_decoder"]:
363+
config = getattr(getattr(ov_pipeline, component), "config", None)
364+
if config is not None:
365+
loaded_config = getattr(getattr(loaded_pipeline, component), "config")
366+
self.assertEqual(config, loaded_config)
362367
self.assertTrue(loaded_pipeline.safety_checker is not None)
363368
self.assertIsInstance(loaded_pipeline.safety_checker, StableDiffusionSafetyChecker)
364369
del loaded_pipeline

0 commit comments

Comments
 (0)