HelloSun commited on
Commit
0bda7e6
·
verified ·
1 Parent(s): e7ab229

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ model_id = "HelloSun/chilloutmix_NiPrunedFp32Fix-openvino"
6
 
7
  # 確保這些是有效的尺寸
8
  HIGH = 2048
9
- WIDTH = 2048
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=30,
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