K00B404 commited on
Commit
c517de9
·
verified ·
1 Parent(s): 31d24f4

Update lora_saver.py

Browse files
Files changed (1) hide show
  1. lora_saver.py +1 -0
lora_saver.py CHANGED
@@ -9,6 +9,7 @@ HF_REPO = "K00B404/LoraStack"
9
 
10
  def download_lora(repo_id, save_dir="loras"):
11
  """Download all files from a LoRA repository."""
 
12
  model_name = repo_id.split("/")[-1]
13
  target_dir = os.path.join(save_dir, model_name)
14
  os.makedirs(target_dir, exist_ok=True)
 
9
 
10
  def download_lora(repo_id, save_dir="loras"):
11
  """Download all files from a LoRA repository."""
12
+ print(f"lora repo {repo_id}")
13
  model_name = repo_id.split("/")[-1]
14
  target_dir = os.path.join(save_dir, model_name)
15
  os.makedirs(target_dir, exist_ok=True)