TalHach61 commited on
Commit
12b69e9
·
verified ·
1 Parent(s): cc620ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -174,6 +174,7 @@ def infer(cond_in, image_in, prompt, inference_steps, guidance_scale, control_mo
174
  num_inference_steps=inference_steps,
175
  guidance_scale=guidance_scale,
176
  generator=torch.manual_seed(seed),
 
177
  ).images[0]
178
 
179
  torch.cuda.empty_cache()
 
174
  num_inference_steps=inference_steps,
175
  guidance_scale=guidance_scale,
176
  generator=torch.manual_seed(seed),
177
+ max_sequence_length=128,
178
  ).images[0]
179
 
180
  torch.cuda.empty_cache()