Upload folder using huggingface_hub

aagenerator.py

@@ -134,7 +134,7 @@ def parse_args():
     parser.add_argument('--step_num2', default=50, type=int)
     parser.add_argument('--size', default=2048, type=int)
     parser.add_argument("--pretrained_vae_model_name_or_path", type=str,
-                        default='stabilityai/sdxl-vae-fp16-fix')
+                        default='madebyollin/sdxl-vae-fp16-fix')
     parser.add_argument('--normalize_dist', default=True, action="store_false")
     parser.add_argument('--change_whole_to_body', default=True, action="store_false")
     parser.add_argument('--off_wa', default=True, action="store_false")