Spaces:

snap-research
/

weights2weights

Running on Zero

amildravid4292 commited on Jul 21, 2024

Commit

94c4f9d

verified ·

1 Parent(s): 79d99bd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -305,7 +305,7 @@ def sample_then_run():
     image = inference( prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.value.proj.detach(), "model.pt" )
     net = torch.load("model.pt").cpu()
-    print(net)
     return image, "model.pt", net   #, network.value.cpu()

     image = inference( prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.value.proj.detach(), "model.pt" )
     net = torch.load("model.pt").cpu()
     return image, "model.pt", net   #, network.value.cpu()