Update lora_saver.py
Browse files- lora_saver.py +1 -0
lora_saver.py
CHANGED
@@ -19,6 +19,7 @@ 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 |
print(f"Downloading {file} from {repo_id.get('repo')}...")
|
23 |
hf_hub_download(repo_id=repo_id.get('repo'), filename=file, local_dir=target_dir)
|
24 |
|
|
|
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 |
|