This commit is contained in:
oobabooga 2023-07-12 13:33:44 -07:00
parent 6447b2eea6
commit c592a9b740

View File

@ -146,11 +146,6 @@ def download_model_wrapper(repo_id, progress=gr.Progress()):
yield ("Getting the output folder") yield ("Getting the output folder")
models_dir = Path(shared.args.model_dir) models_dir = Path(shared.args.model_dir)
# If the last part of the path is "models", remove it
if models_dir.name.lower() == 'models':
models_dir = models_dir.parent
output_folder = downloader.get_output_folder(model, branch, is_lora, base_folder=models_dir) output_folder = downloader.get_output_folder(model, branch, is_lora, base_folder=models_dir)
if check: if check: