path fixed

This commit is contained in:
Usama Kenway 2023-04-09 16:53:21 +05:00
parent 7436dd5b4a
commit ebdf4c8c12

View File

@ -195,9 +195,9 @@ def download_model_wrapper(repo_id):
is_lora = True is_lora = True
repo_dir_name = repo_id.replace("/", "--") repo_dir_name = repo_id.replace("/", "--")
if is_lora is True: if is_lora is True:
models_dir = ".loras" models_dir = "loras"
else: else:
models_dir = ".models" models_dir = "models"
if not os.path.exists(models_dir): if not os.path.exists(models_dir):
os.makedirs(models_dir) os.makedirs(models_dir)
repo_dir = os.path.join(models_dir, repo_dir_name) repo_dir = os.path.join(models_dir, repo_dir_name)