revise download API

app-huggingface.py

@@ -65,7 +65,7 @@ if not LOCAL_DEBUG:
     os.makedirs(DreamBooth_LoRA_PATH, exist_ok=True)
     os.makedirs(STABLE_DIFFUSION_PATH, exist_ok=True)
 
-    hf_hub_download(model_repo='leoxing/PIA-pruned',
+    hf_hub_download(repo_id='leoxing/PIA-pruned',
                     model_name='PIA', output=PIA_PATH)
     os.system('bash download_bashscripts/1-RealisticVision.sh')
     os.system('bash download_bashscripts/2-RcnzCartoon.sh')
@@ -81,18 +81,18 @@ if not LOCAL_DEBUG:
                                       cache_dir='models/StableDiffusion')
 
     # vae
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='vae', filename='config.json',
                     cache_dir='models/StableDiffusion')
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='vae', filename='diffusion_pytorch_model.bin',
                     cache_dir='models/StableDiffusion')
 
     # text encoder
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='text_encoder', filename='config.json',
                     cache_dir='models/StableDiffusion')
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='text_encoder', filename='pytorch_model.bin',
                     cache_dir='models/StableDiffusion')
 
@@ -100,23 +100,23 @@ if not LOCAL_DEBUG:
     hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
                     subfolder='tokenizer', filename='merges.txt',
                     cache_dir='models/StableDiffusion')
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='tokenizer', filename='special_tokens_map.json',
                     cache_dir='models/StableDiffusion')
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='tokenizer', filename='tokenizer_config.json',
                     cache_dir='models/StableDiffusion')
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='tokenizer', filename='vocab.json',
                     cache_dir='models/StableDiffusion')
 
     # scheduler
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
                     subfolder='scheduler', filename='scheduler_config.json',
                     cache_dir='models/StableDiffusion')
 
     # model index
-    hf_hub_download(model_repo='runwayml/stable-diffusion-v1-5', filename='model_index.json',
+    hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5', filename='model_index.json',
                     cache_dir='models/StableDiffusion')
 
 else: