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:
|