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 |
|