Spaces:

snap-research
/

weights2weights

Running on Zero

amildravid4292 commited on Jul 21, 2024

Commit

c920a8b

verified ·

1 Parent(s): 432815b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -94,8 +94,7 @@ def sample_then_run(network, unet):
     cfg = 3.0
     steps = 25
     image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
-    torch.save(network.proj.detach(), "model.pt" )
-    #return
     return image, "model.pt", unet, network

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