add test for LCMs

echarlaix · echarlaix · commit 50507be57559 · 2024-02-12T15:10:08.000+01:00
diff --git a/tests/openvino/test_export.py b/tests/openvino/test_export.py
@@ -39,6 +39,7 @@
     OVStableDiffusionPipeline,
     OVStableDiffusionXLImg2ImgPipeline,
     OVStableDiffusionXLPipeline,
+    OVLatentConsistencyModelPipeline,
 )
 from optimum.intel.openvino.modeling_base import OVBaseModel
 from optimum.utils.save_utils import maybe_load_preprocessors
@@ -61,6 +62,7 @@ class ExportModelTest(unittest.TestCase):
         "stable-diffusion": OVStableDiffusionPipeline,
         "stable-diffusion-xl": OVStableDiffusionXLPipeline,
         "stable-diffusion-xl-refiner": OVStableDiffusionXLImg2ImgPipeline,
+        "latent-consistency": OVLatentConsistencyModelPipeline,
     }
 
     def _openvino_export(

Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,7 @@`
`39`	`39`	`OVStableDiffusionPipeline,`
`40`	`40`	`OVStableDiffusionXLImg2ImgPipeline,`
`41`	`41`	`OVStableDiffusionXLPipeline,`
	`42`	`+ OVLatentConsistencyModelPipeline,`
`42`	`43`	`)`
`43`	`44`	`from optimum.intel.openvino.modeling_base import OVBaseModel`
`44`	`45`	`from optimum.utils.save_utils import maybe_load_preprocessors`
`@@ -61,6 +62,7 @@ class ExportModelTest(unittest.TestCase):`
`61`	`62`	`"stable-diffusion": OVStableDiffusionPipeline,`
`62`	`63`	`"stable-diffusion-xl": OVStableDiffusionXLPipeline,`
`63`	`64`	`"stable-diffusion-xl-refiner": OVStableDiffusionXLImg2ImgPipeline,`
	`65`	`+ "latent-consistency": OVLatentConsistencyModelPipeline,`
`64`	`66`	`}`
`65`	`67`
`66`	`68`	`def _openvino_export(`