Garrett Goon commited on
Commit
5f1ae9e
·
1 Parent(s): f42e32b

guidance scale default and no collected imgs

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -59,10 +59,6 @@ def replace_concept_tokens(text: str):
59
  text = text.replace(concept_token, dummy_tokens)
60
  return text
61
 
62
-
63
- all_imgs = []
64
-
65
-
66
  def inference(prompt: str, guidance_scale: int, num_inference_steps: int, seed: int):
67
  prompt = replace_concept_tokens(prompt)
68
  generator = torch.Generator(device=device).manual_seed(seed)
@@ -76,8 +72,7 @@ def inference(prompt: str, guidance_scale: int, num_inference_steps: int, seed:
76
  filtered_imgs = [
77
  img if not nsfw else NSFW_IMAGE for img, nsfw in zip(img_list, nsfw_list)
78
  ]
79
- all_imgs.extend(filtered_imgs)
80
- return all_imgs
81
 
82
 
83
  css = """
@@ -306,7 +301,7 @@ with block:
306
  label="Steps", minimum=1, maximum=80, value=40, step=1
307
  )
308
  guidance_scale = gr.Slider(
309
- label="Guidance Scale", minimum=1.0, maximum=25.0, value=4.0, step=0.1
310
  )
311
  seed = gr.Slider(
312
  label="Seed",
 
59
  text = text.replace(concept_token, dummy_tokens)
60
  return text
61
 
 
 
 
 
62
  def inference(prompt: str, guidance_scale: int, num_inference_steps: int, seed: int):
63
  prompt = replace_concept_tokens(prompt)
64
  generator = torch.Generator(device=device).manual_seed(seed)
 
72
  filtered_imgs = [
73
  img if not nsfw else NSFW_IMAGE for img, nsfw in zip(img_list, nsfw_list)
74
  ]
75
+ return filtered_imgs
 
76
 
77
 
78
  css = """
 
301
  label="Steps", minimum=1, maximum=80, value=40, step=1
302
  )
303
  guidance_scale = gr.Slider(
304
+ label="Guidance Scale", minimum=1.0, maximum=25.0, value=7.5, step=0.1
305
  )
306
  seed = gr.Slider(
307
  label="Seed",