Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,8 @@ pipe = pipe.to(device)
|
|
| 25 |
|
| 26 |
|
| 27 |
transformer2 = FluxTransformer2DModel.from_pretrained(
|
| 28 |
-
"black-forest-labs/FLUX.1-dev", subfolder="transformer", torch_dtype=dtype
|
|
|
|
| 29 |
)
|
| 30 |
pipe2 = FluxPipeline.from_pretrained(bfl_repo, transformer=None, torch_dtype=dtype)
|
| 31 |
pipe2.transformer = transformer2
|
|
|
|
| 25 |
|
| 26 |
|
| 27 |
transformer2 = FluxTransformer2DModel.from_pretrained(
|
| 28 |
+
"black-forest-labs/FLUX.1-dev", subfolder="transformer", torch_dtype=dtype,
|
| 29 |
+
use_auth_token=os.getenv("HF_TOKEN"),
|
| 30 |
)
|
| 31 |
pipe2 = FluxPipeline.from_pretrained(bfl_repo, transformer=None, torch_dtype=dtype)
|
| 32 |
pipe2.transformer = transformer2
|