amildravid4292 commited on
Commit
87f222b
·
verified ·
1 Parent(s): 18b14c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def sample_then_run(network, unet):
94
  cfg = 3.0
95
  steps = 25
96
  image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
97
- torch.save(network.proj.detach().cpu(), "model.pt" )
98
  return image, "model.pt", unet, network
99
 
100
 
 
94
  cfg = 3.0
95
  steps = 25
96
  image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
97
+ #torch.save(network.proj.detach().cpu(), "model.pt" )
98
  return image, "model.pt", unet, network
99
 
100