fffiloni commited on
Commit
ede6809
1 Parent(s): 2d01911

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def infer(prompt, init_image):
33
  num_inference_steps=50,
34
  generator=generator,
35
  text_embedding_optimization_steps=500,
36
- model_fine_tuning_optimization_steps=1000)
37
 
38
  with torch.no_grad():
39
  torch.cuda.empty_cache()
 
33
  num_inference_steps=50,
34
  generator=generator,
35
  text_embedding_optimization_steps=500,
36
+ model_fine_tuning_optimization_steps=500)
37
 
38
  with torch.no_grad():
39
  torch.cuda.empty_cache()