revise download API

app-huggingface.py

@@ -66,19 +66,19 @@ if not LOCAL_DEBUG:
     os.makedirs(STABLE_DIFFUSION_PATH, exist_ok=True)
 
     hf_hub_download(repo_id='leoxing/PIA-pruned',
-                    model_name='PIA', output=PIA_PATH)
+                    filename='pia.ckpt', output=PIA_PATH)
     os.system('bash download_bashscripts/1-RealisticVision.sh')
     os.system('bash download_bashscripts/2-RcnzCartoon.sh')
     print(os.listdir(DreamBooth_LoRA_PATH))
 
     # unet
     unet_full_path = hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
-                                      subfolder='unet', filename='diffusion_pytorch_model.bin',
-                                      cache_dir='models/StableDiffusion')
+                                     subfolder='unet', filename='diffusion_pytorch_model.bin',
+                                     cache_dir='models/StableDiffusion')
     STABLE_DIFFUSION_PATH = '/'.join(unet_full_path.split('/')[:-2])
     hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
-                                      subfolder='unet', filename='config.json',
-                                      cache_dir='models/StableDiffusion')
+                    subfolder='unet', filename='config.json',
+                    cache_dir='models/StableDiffusion')
 
     # vae
     hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',