XavierJiezou commited on
Commit
6666d79
·
verified ·
1 Parent(s): 5443a64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)