Temporarily disable tokenizer in image generation pipeline

app.py

@@ -25,14 +25,16 @@ def get_image_gen_pipeline():
             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",
+            # tokenizer = AutoTokenizer.from_pretrained(
+            #     "black-forest-labs/FLUX.1-schnell",
+            #     # "black-forest-labs/FLUX.1-dev",
                 use_fast=True
             )
             _image_gen_pipeline = DiffusionPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-schnell",
+                # "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
-                tokenizer=tokenizer
+            #     tokenizer=tokenizer
             ).to(device)
         except Exception as e:
             print(f"Error loading image generation model: {e}")