Spaces:
Runtime error
Runtime error
pengdaqian
commited on
Commit
•
65c3ba9
1
Parent(s):
8c80b14
fix more
Browse files
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 |
|