liyy201912 commited on
Commit
8ba160c
Β·
1 Parent(s): a85c463

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. aagenerator.py +1 -1
aagenerator.py CHANGED
@@ -619,7 +619,7 @@ class Generator:
619
  self.unet_t2mn.requires_grad_(False)
620
 
621
  self.unet = diffusers.UNet2DConditionModel.from_pretrained(
622
- args.pretrained_model_name_or_path2, subfolder="unet", revision=args.revision, use_auth_token=True
623
  )
624
 
625
  # if args.controlnet_model_name_or_path:
 
619
  self.unet_t2mn.requires_grad_(False)
620
 
621
  self.unet = diffusers.UNet2DConditionModel.from_pretrained(
622
+ args.pretrained_model_name_or_path2, subfolder="unet", revision=args.revision, use_auth_token=False
623
  )
624
 
625
  # if args.controlnet_model_name_or_path: