Linoy Tsaban commited on
Commit
9df32ef
1 Parent(s): 87c0592

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -124,7 +124,7 @@ def invert_and_reconstruct(
124
  ):
125
  offsets=(0,0,0,0)
126
  torch.manual_seed(seed)
127
- x0 = load_512(input_image, **offsets, device)
128
 
129
 
130
  # invert
@@ -163,7 +163,7 @@ def edit(input_image,
163
 
164
 
165
  offsets = (0,0,0,0)
166
- x0 = load_512(input_image, **offsets, device)
167
 
168
  # invert
169
  # wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)
 
124
  ):
125
  offsets=(0,0,0,0)
126
  torch.manual_seed(seed)
127
+ x0 = load_512(input_image, device=device)
128
 
129
 
130
  # invert
 
163
 
164
 
165
  offsets = (0,0,0,0)
166
+ x0 = load_512(input_image, device=device)
167
 
168
  # invert
169
  # wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)