Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 10:35:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fix lora download folder

Browse source
This commit is contained in:
oobabooga 2023-07-12 14:19:33 -07:00
parent c592a9b740
commit 0e6295886d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
server.py
View file

@ -145,8 +145,8 @@ 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")
models_dir = Path(shared.args.model_dir)
output_folder = downloader.get_output_folder(model, branch, is_lora, base_folder=models_dir)
base_folder = shared.args.lora_dir if is_lora else shared.args.model_dir
output_folder = downloader.get_output_folder(model, branch, is_lora, base_folder=base_folder)
if check:
progress(0.5)