Spaces:
Runtime error
Runtime error
Garrett Goon
commited on
Commit
·
5f1ae9e
1
Parent(s):
f42e32b
guidance scale default and no collected imgs
Browse files
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 |
-
|
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=
|
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",
|