fffiloni commited on
Commit
542b1e3
1 Parent(s): f95a9d0

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=250)
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=512)
37
 
38
  with torch.no_grad():
39
  torch.cuda.empty_cache()