Spaces:

snap-research
/

weights2weights

Running on Zero

amildravid4292 commited on Jul 21, 2024

Commit

d3a7e42

verified ·

1 Parent(s): 0bc198e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -303,7 +303,9 @@ def sample_then_run():
     cfg = 3.0
     steps = 25
     image = inference( prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.value.proj, "model.pt" )
     return image, "model.pt", network.value.cpu()

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