BILLY12138
commited on
Commit
•
2c19b8d
1
Parent(s):
12ced3a
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
|
|
8 |
import transformers
|
9 |
transformers.utils.move_cache()
|
10 |
|
11 |
-
SAFETY_CHECKER =
|
12 |
|
13 |
loaded_acc = None
|
14 |
device = "cuda"
|
@@ -169,7 +169,7 @@ Target-Driven Distillation (TDD) is a state-of-the-art consistency distillation
|
|
169 |
label="eta",
|
170 |
minimum=0,
|
171 |
maximum=0.3,
|
172 |
-
step=0.
|
173 |
value=0.2,
|
174 |
)
|
175 |
with gr.Row():
|
@@ -196,8 +196,9 @@ Target-Driven Distillation (TDD) is a state-of-the-art consistency distillation
|
|
196 |
gr.Examples(
|
197 |
examples=[
|
198 |
["A photo of a cat made of water.", "", "SDXL-1.0", "TDD_adv", 4, 1.7, 0.2, 546237],
|
199 |
-
["
|
200 |
-
|
|
|
201 |
],
|
202 |
inputs=[prompt, negative_prompt, ckpt, acc, steps, guidance_scale, eta, seed],
|
203 |
outputs=[img],
|
|
|
8 |
import transformers
|
9 |
transformers.utils.move_cache()
|
10 |
|
11 |
+
SAFETY_CHECKER = True
|
12 |
|
13 |
loaded_acc = None
|
14 |
device = "cuda"
|
|
|
169 |
label="eta",
|
170 |
minimum=0,
|
171 |
maximum=0.3,
|
172 |
+
step=0.01,
|
173 |
value=0.2,
|
174 |
)
|
175 |
with gr.Row():
|
|
|
196 |
gr.Examples(
|
197 |
examples=[
|
198 |
["A photo of a cat made of water.", "", "SDXL-1.0", "TDD_adv", 4, 1.7, 0.2, 546237],
|
199 |
+
["Astronaut in a jungle, cold color palette, muted colors, detailed, 8k", "NSFW, low quality, low resolution, normal quality", "Real", "TDD_adv", 7, 2.0, 0.3, 68436]
|
200 |
+
["panda mad scientist mixing sparkling chemicals", "", "Real", "TDD_adv", 5, 1.7, 0.2, 3853],
|
201 |
+
["a dog in snow, 8k", "", "SDXL-1.0", "TDD", 6, 1.3, 0.2, 12138]
|
202 |
],
|
203 |
inputs=[prompt, negative_prompt, ckpt, acc, steps, guidance_scale, eta, seed],
|
204 |
outputs=[img],
|