This commit is contained in:
wassname
2023-04-22 07:50:21 +00:00
parent d149f12704
commit 3410c9cad0
5 changed files with 9 additions and 8 deletions
+1 -1
View File
@@ -164,7 +164,7 @@ def get_download_links_from_huggingface(model, branch, text_only=False):
def get_output_folder(model, branch, is_lora, base_folder=None):
if base_folder is None:
base_folder = 'models' if not is_lora else 'loras'
base_folder = 'data/models' if not is_lora else 'data/loras'
output_folder = f"{'_'.join(model.split('/')[-2:])}"
if branch != 'main':