TheStageAI
/

Elastic-FLUX.1-dev

Model card Files Files and versions Community

quazim commited on May 19

Commit

f5db075

·

verified ·

1 Parent(s): 4143be0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ device = torch.device("cuda")
 pipeline = FluxPipeline.from_pretrained(
     mode_name,
     torch_dtype=torch.bfloat16,
-    token=hf_token
     mode='S'
 )
 pipeline.to(device)

 pipeline = FluxPipeline.from_pretrained(
     mode_name,
     torch_dtype=torch.bfloat16,
+    token=hf_token,
     mode='S'
 )
 pipeline.to(device)