flux-labs

Running on Zero

vilarin commited on Jun 14

Commit

a775243

•

1 Parent(s): 8bfe8b7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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