hsuwill000 commited on
Commit
5cb4c34
·
verified ·
1 Parent(s): eb720c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ model_id = "Disty0/LCM_SoteMix"
14
  #model_id = "Disty0/sotediffusion-v2" #不可
15
 
16
  #1024*512 記憶體不足
17
- HIGH=512
18
  WIDTH=512
19
 
20
  batch_size = -1
@@ -68,7 +68,7 @@ def infer(prompt,negative_prompt):
68
  width = WIDTH,
69
  height = HIGH,
70
  guidance_scale=1.0,
71
- num_inference_steps=4,
72
  num_images_per_prompt=1,
73
  ).images[0]
74
 
 
14
  #model_id = "Disty0/sotediffusion-v2" #不可
15
 
16
  #1024*512 記憶體不足
17
+ HIGH=1024
18
  WIDTH=512
19
 
20
  batch_size = -1
 
68
  width = WIDTH,
69
  height = HIGH,
70
  guidance_scale=1.0,
71
+ num_inference_steps=8,
72
  num_images_per_prompt=1,
73
  ).images[0]
74