amildravid4292 commited on
Commit
79d99bd
·
verified ·
1 Parent(s): 866842c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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