fix more

model.py

@@ -3,7 +3,6 @@ from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler, DPMSolver
     OnnxStableDiffusionPipeline
 
 import pipeline_openvino_stable_diffusion
-from optimum.intel.openvino import OVStableDiffusionPipeline
 
 
 def get_sd_21():
@@ -31,19 +30,40 @@ def get_sd_21():
 def get_sd_every():
     model_id = 'OFA-Sys/small-stable-diffusion-v0'
     scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
-    pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/stable-diffusion-2-1-quantized", compile=False)
+
+    onnx_pipe = OnnxStableDiffusionPipeline.from_pretrained(
+        "OFA-Sys/small-stable-diffusion-v0",
+        scheduler=scheduler,
+        revision="onnx",
+        provider="CPUExecutionProvider",
+    )
+    pipe = pipeline_openvino_stable_diffusion.OpenVINOStableDiffusionPipeline.from_onnx_pipeline(onnx_pipe)
     return pipe
 
 
 def get_sd_small():
-    pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/stable-diffusion-2-1-quantized", compile=False)
-    pipe.reshape(batch_size=1, height=256, width=256, num_images_per_prompt=1)
-    pipe.compile()
+    model_id = 'OFA-Sys/small-stable-diffusion-v0'
+    scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
+
+    onnx_pipe = OnnxStableDiffusionPipeline.from_pretrained(
+        "OFA-Sys/small-stable-diffusion-v0",
+        scheduler=scheduler,
+        revision="onnx",
+        provider="CPUExecutionProvider",
+    )
+    pipe = pipeline_openvino_stable_diffusion.OpenVINOStableDiffusionPipeline.from_onnx_pipeline(onnx_pipe)
     return pipe
 
 
 def get_sd_tiny():
-    pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/stable-diffusion-2-1-quantized", compile=False)
-    pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
-    pipe.compile()
+    model_id = 'OFA-Sys/small-stable-diffusion-v0'
+    scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
+
+    onnx_pipe = OnnxStableDiffusionPipeline.from_pretrained(
+        "OFA-Sys/small-stable-diffusion-v0",
+        scheduler=scheduler,
+        revision="onnx",
+        provider="CPUExecutionProvider",
+    )
+    pipe = pipeline_openvino_stable_diffusion.OpenVINOStableDiffusionPipeline.from_onnx_pipeline(onnx_pipe)
     return pipe

requirements.txt

@@ -8,5 +8,4 @@ onnx
 openvino
 onnxruntime-openvino
 ftfy
-py-cpuinfo
-optimum-intel[openvino, diffusers]
+py-cpuinfo