sts reduced flux max seq length to 128

2024-08-10 20:39:17 -06:00 · 2024-08-10 20:39:17 -06:00 · bae3e88d2e
commit bae3e88d2e
parent 2ca883bb77
1 changed files with 1 additions and 1 deletions
--- a/flux-schnell.py
+++ b/flux-schnell.py
@ -36,7 +36,7 @@ output = pipe(
    width=width,
    num_images_per_prompt=4,
    num_inference_steps=4,
-    max_sequence_length=256,
+    max_sequence_length=128,
    guidance_scale=0.0,
 )
 print('Max mem allocated (GB) while denoising:', torch.cuda.max_memory_allocated() / (1024 ** 3))