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

Update lora_saver.py

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