flux-labs

Running on Zero

vilarin commited on Jun 14, 2024

Commit

8bfe8b7

verified ·

1 Parent(s): 0f370cf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,6 +104,7 @@ def generate_image(
             guidance_scale=scale,
             num_inference_steps=steps,
             generator = generator,
         )
     else:
         image = pipe(
@@ -114,7 +115,7 @@ def generate_image(
             guidance_scale=scale,
             num_inference_steps=steps,
             generator = generator,
-            max_length=512,
         )
     return image.images[0]

             guidance_scale=scale,
             num_inference_steps=steps,
             generator = generator,
+            tokenizer_max_length = 512,
         )
     else:
         image = pipe(
             guidance_scale=scale,
             num_inference_steps=steps,
             generator = generator,
+            tokenizer_max_length = 512,
         )
     return image.images[0]