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='/fsx_laion/alvin/pretrain/sdxl-vae-fp16-fix')
+                        default='stabilityai/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")
@@ -169,9 +169,9 @@ def parse_args():
     parser.add_argument('--start', default=0, type=int)
     parser.add_argument('--end', default=8236, type=int)
     parser.add_argument("--pretrained_model_name_or_path", type=str,
-                        default='/fsx_laion/alvin/pretrain/stable-diffusion-2-base')
+                        default='stabilityai/stable-diffusion-2-base')
     parser.add_argument("--pretrained_model_name_or_path2", type=str,
-                        default='/fsx_laion/alvin/pretrain/stable-diffusion-xl-base-1.0')
+                        default='stabilityai/stable-diffusion-xl-base-1.0')
     parser.add_argument('--prediction_type', type=str, default='v_prediction',
                         choices=['epsilon', 'v_prediction', 'target'], help='Select a mode')
     parser.add_argument('--prediction_type2', type=str, default='epsilon',
@@ -783,7 +783,7 @@ class Generator:
         self.pipeline2.set_progress_bar_config(disable=False)
 
         self.refiner = DiffusionPipeline.from_pretrained(
-            "/fsx_laion/alvin/pretrain/stable-diffusion-xl-refiner-1.0",
+            "stabilityai/stable-diffusion-xl-refiner-1.0",
             text_encoder_2=self.pipeline2.text_encoder_2,
             vae=self.pipeline2.vae,
             torch_dtype=torch.float16,