Linoy Tsaban commited on
Commit
7ff8616
1 Parent(s): 9156300

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -129,12 +129,12 @@ def invert_and_reconstruct(
129
 
130
  if do_inversion:
131
  # invert and retrieve noise maps and latent
132
- zs_tensor, wts_tensor = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=cfg_scale_src)
133
  wts = gr.State(value=wts_tensor)
134
  zs = gr.State(value=zs_tensor)
135
  do_inversion = False
136
 
137
- output = sample(zs.value, wts.value, prompt_tar=tar_prompt, skip=skip, cfg_scale_tar=cfg_scale_tar)
138
 
139
  return output, wts, zs, do_inversion
140
 
 
129
 
130
  if do_inversion:
131
  # invert and retrieve noise maps and latent
132
+ zs_tensor, wts_tensor = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)
133
  wts = gr.State(value=wts_tensor)
134
  zs = gr.State(value=zs_tensor)
135
  do_inversion = False
136
 
137
+ output = sample(zs.value, wts.value, prompt_tar=tar_prompt, skip=skip, cfg_scale_tar=tar_cfg_scale)
138
 
139
  return output, wts, zs, do_inversion
140