LeoXing1996 commited on
Commit
3317194
1 Parent(s): 0647ed2

revise download API

Browse files
Files changed (1) hide show
  1. app-huggingface.py +5 -5
app-huggingface.py CHANGED
@@ -66,19 +66,19 @@ if not LOCAL_DEBUG:
66
  os.makedirs(STABLE_DIFFUSION_PATH, exist_ok=True)
67
 
68
  hf_hub_download(repo_id='leoxing/PIA-pruned',
69
- model_name='PIA', output=PIA_PATH)
70
  os.system('bash download_bashscripts/1-RealisticVision.sh')
71
  os.system('bash download_bashscripts/2-RcnzCartoon.sh')
72
  print(os.listdir(DreamBooth_LoRA_PATH))
73
 
74
  # unet
75
  unet_full_path = hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
76
- subfolder='unet', filename='diffusion_pytorch_model.bin',
77
- cache_dir='models/StableDiffusion')
78
  STABLE_DIFFUSION_PATH = '/'.join(unet_full_path.split('/')[:-2])
79
  hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
80
- subfolder='unet', filename='config.json',
81
- cache_dir='models/StableDiffusion')
82
 
83
  # vae
84
  hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
 
66
  os.makedirs(STABLE_DIFFUSION_PATH, exist_ok=True)
67
 
68
  hf_hub_download(repo_id='leoxing/PIA-pruned',
69
+ filename='pia.ckpt', output=PIA_PATH)
70
  os.system('bash download_bashscripts/1-RealisticVision.sh')
71
  os.system('bash download_bashscripts/2-RcnzCartoon.sh')
72
  print(os.listdir(DreamBooth_LoRA_PATH))
73
 
74
  # unet
75
  unet_full_path = hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
76
+ subfolder='unet', filename='diffusion_pytorch_model.bin',
77
+ cache_dir='models/StableDiffusion')
78
  STABLE_DIFFUSION_PATH = '/'.join(unet_full_path.split('/')[:-2])
79
  hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',
80
+ subfolder='unet', filename='config.json',
81
+ cache_dir='models/StableDiffusion')
82
 
83
  # vae
84
  hf_hub_download(repo_id='runwayml/stable-diffusion-v1-5',