Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -303,7 +303,7 @@ def sample_then_run():
|
|
303 |
cfg = 3.0
|
304 |
steps = 25
|
305 |
image = inference( prompt, negative_prompt, cfg, steps, seed)
|
306 |
-
torch.save(network.value.proj, "model.pt" )
|
307 |
net = torch.load("model.pt").cpu()
|
308 |
print(net)
|
309 |
|
|
|
303 |
cfg = 3.0
|
304 |
steps = 25
|
305 |
image = inference( prompt, negative_prompt, cfg, steps, seed)
|
306 |
+
torch.save(network.value.proj.detach(), "model.pt" )
|
307 |
net = torch.load("model.pt").cpu()
|
308 |
print(net)
|
309 |
|