Spaces:

editing-images
/

ledits

Running on A10G

Linoy Tsaban commited on May 25, 2023

Commit

dd848f1

•

1 Parent(s): a5fdee9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,8 +55,7 @@ def edit(input_image, input_image_prompt='', target_prompt='', edit_prompt='',
          edit_guidance_scale=8, guidance_scale=15, skip=36, num_diffusion_steps=100,
          ):
     offsets=(0,0,0,0)
-    input_image = pil_to_tensor(input_image).type(torch.float16)
-    x0 = load_512(input_image, *offsets, device)
     # invert

          edit_guidance_scale=8, guidance_scale=15, skip=36, num_diffusion_steps=100,
          ):
     offsets=(0,0,0,0)
+    x0 = pil_to_tensor(load_512(input_image, *offsets, device)).type(torch.float16)
     # invert