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
repo_dir_name = repo_id.replace("/", "--")
if is_lora is True:
models_dir = ".loras"
models_dir = "loras"
else:
models_dir = ".models"
models_dir = "models"
if not os.path.exists(models_dir):
os.makedirs(models_dir)
repo_dir = os.path.join(models_dir, repo_dir_name)