Spaces:

superbsaeed
/

text2image

Sleeping

superbsaeed commited on Apr 13

Commit

5099d05

verified ·

1 Parent(s): b6faa80

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ model_id = "black-forest-labs/FLUX.1-schnell"
 # Load the model with bfloat16 to save memory and improve speed
 pipe = FluxPipeline.from_pretrained(
     model_id,
-    torch_dtype=torch.bfloat16
     hf_token = os.getenv("FLUX_API_TOKEN")
 )

 # Load the model with bfloat16 to save memory and improve speed
 pipe = FluxPipeline.from_pretrained(
     model_id,
+    torch_dtype=torch.bfloat16,
     hf_token = os.getenv("FLUX_API_TOKEN")
 )