Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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()
|