Spaces:

snap-research
/

weights2weights

Running on Zero

amildravid4292 commited on Jul 21, 2024

Commit

8f9ef3b

verified ·

1 Parent(s): df5fc9d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -96,7 +96,7 @@ def sample_then_run(network, unet):
     image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.proj.detach().cpu(), "model.pt" )
     print("done saving")
-    return image, "model.pt", unet, network.cpu()

     image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.proj.detach().cpu(), "model.pt" )
     print("done saving")
+    return image, "model.pt", unet.cpu(), network.cpu()