Spaces:

snap-research
/

weights2weights

Running on Zero

linoyts HF Staff commited on Jul 16, 2024

Commit

d6adfd6

verified ·

1 Parent(s): fdd9574

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -360,7 +360,7 @@ def run_inversion(input_image, pcs, epochs, weight_decay,lr):
     steps = 50
     image = inference( prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.proj, "model.pt" )
-    return image, "model.pt"

     steps = 50
     image = inference( prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.proj, "model.pt" )
+    return (image,init_image), "model.pt"