Update README.md
Browse files
README.md
CHANGED
|
@@ -52,7 +52,7 @@ device = torch.device("cuda")
|
|
| 52 |
pipeline = FluxPipeline.from_pretrained(
|
| 53 |
mode_name,
|
| 54 |
torch_dtype=torch.bfloat16,
|
| 55 |
-
token=hf_token
|
| 56 |
mode='S'
|
| 57 |
)
|
| 58 |
pipeline.to(device)
|
|
|
|
| 52 |
pipeline = FluxPipeline.from_pretrained(
|
| 53 |
mode_name,
|
| 54 |
torch_dtype=torch.bfloat16,
|
| 55 |
+
token=hf_token,
|
| 56 |
mode='S'
|
| 57 |
)
|
| 58 |
pipeline.to(device)
|