K00B404 commited on
Commit
4744490
·
verified ·
1 Parent(s): 9e16bb6

Update lora_saver.py

Browse files
Files changed (1) hide show
  1. 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
- 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
 
 
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