tokenizer flux

app.py

@@ -24,9 +24,15 @@ def get_image_gen_pipeline():
         try:
             device = "cuda" if torch.cuda.is_available() else "cpu"
             dtype = torch.bfloat16
+            # Load fast tokenizer for the image pipeline
+            tokenizer = AutoTokenizer.from_pretrained(
+                "black-forest-labs/FLUX.1-schnell",
+                use_fast=True
+            )
             _image_gen_pipeline = DiffusionPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-schnell",
-                torch_dtype=dtype
+                torch_dtype=dtype,
+                tokenizer=tokenizer
             ).to(device)
         except Exception as e:
             print(f"Error loading image generation model: {e}")
@@ -295,8 +301,11 @@ def create_interface():
                     )
             
             gr.Examples(
-                examples=examples,
-                fn=infer,
+                examples=[
+                    "a backpack for kids, flower style",
+                    "medieval flip flops",
+                    "cat shaped cake mold"
+                ],
                 inputs=[prompt],
                 outputs=[generated_image, seed],
                 cache_examples=True,