Update download.py
Browse files- 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('
|
50 |
-
os.makedirs('
|
51 |
for filename in filename_list:
|
52 |
-
local_file = os.path.join('
|
53 |
|
54 |
if not os.path.exists(local_file):
|
55 |
-
hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='
|
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):
|