mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-22 08:07:56 +01:00
parent
3778816b8d
commit
3708de2b1f
@ -145,7 +145,13 @@ def download_model_wrapper(repo_id, progress=gr.Progress()):
|
||||
links, sha256, is_lora = downloader.get_download_links_from_huggingface(model, branch, text_only=False)
|
||||
|
||||
yield ("Getting the output folder")
|
||||
output_folder = downloader.get_output_folder(model, branch, is_lora)
|
||||
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)
|
||||
|
||||
if check:
|
||||
progress(0.5)
|
||||
|
Loading…
Reference in New Issue
Block a user