Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -147,7 +147,7 @@ def initialize_models(args):
|
|
147 |
args.flux_path,
|
148 |
transformer=model.denoise_tower.denoiser,
|
149 |
torch_dtype=torch.float16,
|
150 |
-
|
151 |
)
|
152 |
if args.offload:
|
153 |
pipe.enable_model_cpu_offload()
|
|
|
147 |
args.flux_path,
|
148 |
transformer=model.denoise_tower.denoiser,
|
149 |
torch_dtype=torch.float16,
|
150 |
+
token=os.environ["HF_TOKEN"],
|
151 |
)
|
152 |
if args.offload:
|
153 |
pipe.enable_model_cpu_offload()
|