HelloSun commited on
Commit
36f329a
·
verified ·
1 Parent(s): 9cf3608

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,8 +12,8 @@ model_id = "Disty0/LCM_SoteMix"
12
  #model_id = "Disty0/sotediffusion-v2" #不可
13
 
14
  #1024*512 記憶體不足
15
- HIGH=256
16
- WIDTH=256
17
 
18
 
19
  batch_size = -1
@@ -49,7 +49,7 @@ def infer(prompt,negative_prompt):
49
  negative_prompt = negative_prompt,
50
  width = HIGH,
51
  height = WIDTH,
52
- guidance_scale=7.5,
53
  num_inference_steps=4,
54
  num_images_per_prompt=1,
55
  ).images[0]
 
12
  #model_id = "Disty0/sotediffusion-v2" #不可
13
 
14
  #1024*512 記憶體不足
15
+ HIGH=512
16
+ WIDTH=512
17
 
18
 
19
  batch_size = -1
 
49
  negative_prompt = negative_prompt,
50
  width = HIGH,
51
  height = WIDTH,
52
+ guidance_scale=2.0,
53
  num_inference_steps=4,
54
  num_images_per_prompt=1,
55
  ).images[0]