BILLY12138 commited on
Commit
2c19b8d
1 Parent(s): 12ced3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  import transformers
9
  transformers.utils.move_cache()
10
 
11
- SAFETY_CHECKER = False
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.1,
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
- ["A photo of a dog made of water.", "", "SDXL-1.0", "TDD_adv", 4, 1.7, 0.2, 546237],
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],