hsuwill000 commited on
Commit
e229282
Β·
verified Β·
1 Parent(s): 260a90d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ from typing import Optional, Dict
10
 
11
 
12
 
13
- model_id = "Disty0/LCM_SoteMix"
14
  #model_id = "Disty0/sotediffusion-v2" #不可
15
 
16
  #1024*512 θ¨˜ζ†Άι«”δΈθΆ³
@@ -68,7 +68,7 @@ def infer(prompt,negative_prompt):
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
 
 
10
 
11
 
12
 
13
+ model_id = "stabilityai/sd-turbo"
14
  #model_id = "Disty0/sotediffusion-v2" #不可
15
 
16
  #1024*512 θ¨˜ζ†Άι«”δΈθΆ³
 
68
  width = WIDTH,
69
  height = HIGH,
70
  guidance_scale=1.0,
71
+ num_inference_steps=2,
72
  num_images_per_prompt=1,
73
  ).images[0]
74