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],
|