Spaces:

xilluill
/

KV-Edit

Running on Zero

xilluill commited on Feb 26

Commit

95775e7

1 Parent(s): 9c8bfe6

update resize

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,8 +102,8 @@ def edit(brush_canvas,
     torch.cuda.empty_cache()
     rgba_init_image = brush_canvas["background"]
     init_image = rgba_init_image[:,:,:3]
-    init_image = resize_image(init_image)
     shape = init_image.shape
     height = shape[0] if shape[0] % 16 == 0 else shape[0] - shape[0] % 16
     width = shape[1] if shape[1] % 16 == 0 else shape[1] - shape[1] % 16

     torch.cuda.empty_cache()
     rgba_init_image = brush_canvas["background"]
+    rgba_init_image = resize_image(rgba_init_image)
     init_image = rgba_init_image[:,:,:3]
     shape = init_image.shape
     height = shape[0] if shape[0] % 16 == 0 else shape[0] - shape[0] % 16
     width = shape[1] if shape[1] % 16 == 0 else shape[1] - shape[1] % 16