Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
·
f403525
1
Parent(s):
8e8b324
Update app.py
Browse files
app.py
CHANGED
@@ -154,7 +154,7 @@ with gr.Blocks(css='style.css') as demo:
|
|
154 |
cfg_scale_tar = 15,
|
155 |
skip=36,
|
156 |
seed = 0,
|
157 |
-
|
158 |
|
159 |
# if randomized_seed:
|
160 |
# seed = random.randint(0, np.iinfo(np.int32).max)
|
@@ -163,7 +163,7 @@ with gr.Blocks(css='style.css') as demo:
|
|
163 |
# # offsets=(0,0,0,0)
|
164 |
x0 = load_512(input_image, device=device)
|
165 |
|
166 |
-
if do_inversion or
|
167 |
zs_tensor, wts_tensor = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=cfg_scale_src)
|
168 |
wts = gr.State(value=wts_tensor)
|
169 |
zs = gr.State(value=zs_tensor)
|
@@ -220,7 +220,7 @@ with gr.Blocks(css='style.css') as demo:
|
|
220 |
cfg_scale_src,
|
221 |
cfg_scale_tar,
|
222 |
skip,
|
223 |
-
seed,
|
224 |
],
|
225 |
outputs=[output_image, wts, zs, do_inversion],
|
226 |
)
|
|
|
154 |
cfg_scale_tar = 15,
|
155 |
skip=36,
|
156 |
seed = 0,
|
157 |
+
randomize_seed = True):
|
158 |
|
159 |
# if randomized_seed:
|
160 |
# seed = random.randint(0, np.iinfo(np.int32).max)
|
|
|
163 |
# # offsets=(0,0,0,0)
|
164 |
x0 = load_512(input_image, device=device)
|
165 |
|
166 |
+
if do_inversion or randomize_seed:
|
167 |
zs_tensor, wts_tensor = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=cfg_scale_src)
|
168 |
wts = gr.State(value=wts_tensor)
|
169 |
zs = gr.State(value=zs_tensor)
|
|
|
220 |
cfg_scale_src,
|
221 |
cfg_scale_tar,
|
222 |
skip,
|
223 |
+
seed,randomize_seed
|
224 |
],
|
225 |
outputs=[output_image, wts, zs, do_inversion],
|
226 |
)
|