Spaces:

Cicici1109
/

IEAP

Runtime error

Cicici1109 commited on Jun 2

Commit

3881106

verified ·

1 Parent(s): e7787e7

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -38,7 +38,7 @@ def init_flux_pipeline():
         if not token:
             raise ValueError("HF_TOKEN environment variable not set.")
         pipe = FluxPipeline.from_pretrained(
-            "black-forest-labs/FLUX.1-dev",
             use_auth_token=token,
             torch_dtype=torch.bfloat16
         )

         if not token:
             raise ValueError("HF_TOKEN environment variable not set.")
         pipe = FluxPipeline.from_pretrained(
+            "black-forest-labs/FLUX.1-schnell",
             use_auth_token=token,
             torch_dtype=torch.bfloat16
         )