app

app.py

@@ -61,15 +61,19 @@ def init_pipeline_if_needed():
             "sayakpaul/flux.1-schell-int8wo-improved",
             torch_dtype=torch.bfloat16,
             use_safetensors=False,
+            token=hf_token,
         )
         _pipe = FluxPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-schnell",
             transformer=transformer_model,
             torch_dtype=torch.bfloat16,
+            token=hf_token,
         )
     else:
         _pipe = FluxPipeline.from_pretrained(
-            "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16
+            "black-forest-labs/FLUX.1-schnell",
+            torch_dtype=torch.bfloat16,
+            token=hf_token,
         )
 
     _pipe = _pipe.to("cuda")