fffiloni commited on
Commit
7b7bde3
1 Parent(s): 5e59dc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,8 +28,8 @@ def infer(prompt, init_image):
28
  guidance_scale=7.5,
29
  num_inference_steps=50,
30
  generator=generator,
31
- text_embedding_optimization_steps: 100,
32
- model_fine_tuning_optimization_steps: 500)
33
 
34
  with torch.no_grad():
35
  torch.cuda.empty_cache()
 
28
  guidance_scale=7.5,
29
  num_inference_steps=50,
30
  generator=generator,
31
+ text_embedding_optimization_steps=100,
32
+ model_fine_tuning_optimization_steps=500)
33
 
34
  with torch.no_grad():
35
  torch.cuda.empty_cache()