Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -712,7 +712,7 @@ class ImageGenerator:
|
|
712 |
|
713 |
def get_model(self, args):
|
714 |
pipeline = FluxPipeline.from_pretrained(
|
715 |
-
args.pretrained_ckpt, torch_dtype=torch.bfloat16
|
716 |
)
|
717 |
pipeline.load_lora_weights(args.lora_ckpt, weight_name=f"pytorch_lora_weights.safetensors",)
|
718 |
pipeline.to(args.device)
|
|
|
712 |
|
713 |
def get_model(self, args):
|
714 |
pipeline = FluxPipeline.from_pretrained(
|
715 |
+
args.pretrained_ckpt, torch_dtype=torch.bfloat16,use_auth_token=True # 这一行很关键
|
716 |
)
|
717 |
pipeline.load_lora_weights(args.lora_ckpt, weight_name=f"pytorch_lora_weights.safetensors",)
|
718 |
pipeline.to(args.device)
|