Fucius commited on
Commit
decd8ea
1 Parent(s): 72c5c13

Update download.py

Browse files
Files changed (1) hide show
  1. download.py +4 -4
download.py CHANGED
@@ -46,13 +46,13 @@ class OMG_download():
46
  def download_controlNet(self):
47
  REPO_ID = 'lllyasviel/ControlNet'
48
  filename_list = ['annotator/ckpts/body_pose_model.pth']
49
- if not os.path.exists('./checkpoints/ControlNet/'):
50
- os.makedirs('./checkpoints/ControlNet/')
51
  for filename in filename_list:
52
- local_file = os.path.join('./checkpoints/ControlNet/', filename)
53
 
54
  if not os.path.exists(local_file):
55
- hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/ControlNet/',
56
  local_dir_use_symlinks=False)
57
 
58
  def download_characters(self):
 
46
  def download_controlNet(self):
47
  REPO_ID = 'lllyasviel/ControlNet'
48
  filename_list = ['annotator/ckpts/body_pose_model.pth']
49
+ if not os.path.exists('/home/user/app/checkpoint/ControlNet/'):
50
+ os.makedirs('/home/user/app/checkpoint/ControlNet/')
51
  for filename in filename_list:
52
+ local_file = os.path.join('/home/user/app/checkpoint/ControlNet/', filename)
53
 
54
  if not os.path.exists(local_file):
55
+ hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='/home/user/app/checkpoint/ControlNet/',
56
  local_dir_use_symlinks=False)
57
 
58
  def download_characters(self):