Update lora_saver.py
Browse files- lora_saver.py +3 -3
lora_saver.py
CHANGED
@@ -19,9 +19,9 @@ def download_lora(repo_id, save_dir="loras"):
|
|
19 |
files = api.list_repo_files(repo_id.get('repo'), repo_type="model")
|
20 |
|
21 |
for file in files:
|
22 |
-
if not file.endswith('.png') and not file.endswith('.jpg') and not file.endswith('.webp')
|
23 |
-
|
24 |
-
|
25 |
|
26 |
return target_dir
|
27 |
|
|
|
19 |
files = api.list_repo_files(repo_id.get('repo'), repo_type="model")
|
20 |
|
21 |
for file in files:
|
22 |
+
if not file.endswith('.png') and not file.endswith('.jpg') and not file.endswith('.webp'):
|
23 |
+
print(f"Downloading {file} from {repo_id.get('repo')}...")
|
24 |
+
hf_hub_download(repo_id=repo_id.get('repo'), filename=file, local_dir=target_dir)
|
25 |
|
26 |
return target_dir
|
27 |
|