fix more

app.py

@@ -12,7 +12,7 @@ from i18n import i18nTranslator
 word_list_dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
 word_list = word_list_dataset["train"]['Prompt']
 
-from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler, DDIMScheduler, KDPM2AncestralDiscreteScheduler, \
+from diffusers import EulerDiscreteScheduler, DDIMScheduler, KDPM2AncestralDiscreteScheduler, \
     UniPCMultistepScheduler, DPMSolverSinglestepScheduler, DEISMultistepScheduler, PNDMScheduler, \
     DPMSolverMultistepScheduler, HeunDiscreteScheduler, EulerAncestralDiscreteScheduler, DDPMScheduler, \
     LMSDiscreteScheduler, KDPM2DiscreteScheduler

model.py

@@ -32,13 +32,11 @@ def get_sd_small():
     model_id = 'OFA-Sys/small-stable-diffusion-v0'
     scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
 
-    onnx_pipe = OnnxStableDiffusionPipeline.from_pretrained(
+    pipe = OVStableDiffusionPipeline.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)
+    pipe.compile()
     return pipe
 
 

pipeline_openvino_stable_diffusion.py

@@ -159,8 +159,7 @@ class OpenVINOStableDiffusionPipeline(DiffusionPipeline):
                                         device_name="CPU")
 
         # Unet
-        unet_onnx = ie.read_model(
-            model=os.path.join(self.unet.model_save_dir, "model.onnx"))
+        unet_onnx = ie.read_model(model=os.path.join(self.unet.model_save_dir, "model.onnx"))
         unet = ie.compile_model(model=unet_onnx, device_name="CPU")
 
         self.register_modules(vae_decoder=vae_decoder,