Spaces:
Runtime error
Runtime error
Commit
Β·
8ba160c
1
Parent(s):
a85c463
Upload folder using huggingface_hub
Browse files- 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=
|
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:
|