Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,8 @@ model_id = "Disty0/LCM_SoteMix"
|
|
12 |
#model_id = "Disty0/sotediffusion-v2" #不可
|
13 |
|
14 |
#1024*512 記憶體不足
|
15 |
-
HIGH=
|
16 |
-
WIDTH=
|
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=
|
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]
|