Update lora_saver.py
Browse files- lora_saver.py +2 -2
lora_saver.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import os
|
2 |
import shutil
|
3 |
from huggingface_hub import hf_hub_download, upload_folder, HfApi
|
4 |
-
from loras import loras
|
5 |
|
6 |
# Your Hugging Face model repo
|
7 |
HF_REPO = "K00B404/LoraStack"
|
@@ -30,7 +30,7 @@ def upload_to_hf(local_dir, hf_repo):
|
|
30 |
print(f"Uploading {local_dir} to {hf_repo}...")
|
31 |
upload_folder(folder_path=local_dir, repo_id=hf_repo, repo_type="model")
|
32 |
|
33 |
-
def main():
|
34 |
"""Fetch, download, and upload LoRAs to Hugging Face."""
|
35 |
for repo_id in loras:
|
36 |
local_dir = download_lora(repo_id)
|
|
|
1 |
import os
|
2 |
import shutil
|
3 |
from huggingface_hub import hf_hub_download, upload_folder, HfApi
|
4 |
+
#from loras import loras
|
5 |
|
6 |
# Your Hugging Face model repo
|
7 |
HF_REPO = "K00B404/LoraStack"
|
|
|
30 |
print(f"Uploading {local_dir} to {hf_repo}...")
|
31 |
upload_folder(folder_path=local_dir, repo_id=hf_repo, repo_type="model")
|
32 |
|
33 |
+
def main(loras):
|
34 |
"""Fetch, download, and upload LoRAs to Hugging Face."""
|
35 |
for repo_id in loras:
|
36 |
local_dir = download_lora(repo_id)
|