Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ model_id = "HelloSun/chilloutmix_NiPrunedFp32Fix-openvino"
|
|
| 6 |
|
| 7 |
# 確保這些是有效的尺寸
|
| 8 |
HIGH = 2048
|
| 9 |
-
WIDTH =
|
| 10 |
|
| 11 |
pipe = OVStableDiffusionPipeline.from_pretrained(model_id)
|
| 12 |
|
|
@@ -19,7 +19,7 @@ def infer(prompt, negative_prompt):
|
|
| 19 |
width=WIDTH, # 使用 WIDTH
|
| 20 |
height=HIGH, # 使用 HIGH
|
| 21 |
guidance_scale=7.5,
|
| 22 |
-
num_inference_steps=
|
| 23 |
num_images_per_prompt=1,
|
| 24 |
).images[0]
|
| 25 |
return image
|
|
|
|
| 6 |
|
| 7 |
# 確保這些是有效的尺寸
|
| 8 |
HIGH = 2048
|
| 9 |
+
WIDTH = 1024
|
| 10 |
|
| 11 |
pipe = OVStableDiffusionPipeline.from_pretrained(model_id)
|
| 12 |
|
|
|
|
| 19 |
width=WIDTH, # 使用 WIDTH
|
| 20 |
height=HIGH, # 使用 HIGH
|
| 21 |
guidance_scale=7.5,
|
| 22 |
+
num_inference_steps=28,
|
| 23 |
num_images_per_prompt=1,
|
| 24 |
).images[0]
|
| 25 |
return image
|