Texttra
/

Cityscape_Studio

template:diffusion-lora

Model card Files Files and versions

Texttra commited on Apr 7, 2025

Commit

4f82630

·

verified ·

1 Parent(s): c0e327a

Update handler.py

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -12,6 +12,7 @@ class EndpointHandler:
             "black-forest-labs/FLUX.1-dev",
             torch_dtype=torch.float16,
             variant="fp16",
         )
         # Load your LoRA weights from the repo

             "black-forest-labs/FLUX.1-dev",
             torch_dtype=torch.float16,
             variant="fp16",
+            use_auth_token=True
         )
         # Load your LoRA weights from the repo