amildravid4292 commited on
Commit
a205086
·
verified ·
1 Parent(s): 00275ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -396,7 +396,7 @@ def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
396
  cfg = 3.0
397
  steps = 25
398
  image = inference( net, prompt, negative_prompt, cfg, steps, seed)
399
- torch.save(network.proj, "model.pt" )
400
  return net, net, image
401
 
402
 
 
396
  cfg = 3.0
397
  steps = 25
398
  image = inference( net, prompt, negative_prompt, cfg, steps, seed)
399
+
400
  return net, net, image
401
 
402