Linoy Tsaban commited on
Commit
f403525
1 Parent(s): 8e8b324

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- randomized_seed = True):
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 randomized_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,7 +220,7 @@ with gr.Blocks(css='style.css') as demo:
220
  cfg_scale_src,
221
  cfg_scale_tar,
222
  skip,
223
- seed,randomized_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
  )