pengdaqian commited on
Commit
65c3ba9
1 Parent(s): 8c80b14
Files changed (1) hide show
  1. model.py +1 -5
model.py CHANGED
@@ -31,11 +31,7 @@ def get_sd_21():
31
  def get_sd_small():
32
  model_id = 'OFA-Sys/small-stable-diffusion-v0'
33
  scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
34
-
35
- pipe = OVStableDiffusionPipeline.from_pretrained(
36
- "OFA-Sys/small-stable-diffusion-v0",
37
- scheduler=scheduler,
38
- )
39
  pipe.compile()
40
  return pipe
41
 
 
31
  def get_sd_small():
32
  model_id = 'OFA-Sys/small-stable-diffusion-v0'
33
  scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
34
+ pipe = OVStableDiffusionPipeline.from_pretrained("OFA-Sys/small-stable-diffusion-v0", compile=False)
 
 
 
 
35
  pipe.compile()
36
  return pipe
37