Spaces:

snap-research
/

weights2weights

Sleeping

linoyts HF staff commited on Jul 16

Commit

d6adfd6

•

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"