Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-01 11:54:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

More descriptive download info

Browse Source
This commit is contained in:
oobabooga 2023-04-10 11:42:12 -03:00
parent 7e70741a4e
commit 45244ed125
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@ -24,6 +24,7 @@ from modules.LoRA import add_lora_to_model
from modules.models import load_model, load_soft_prompt, unload_model from modules.models import load_model, load_soft_prompt, unload_model
from modules.text_generation import generate_reply, stop_everything_event from modules.text_generation import generate_reply, stop_everything_event
# Loading custom settings # Loading custom settings
settings_file = None settings_file = None
if shared.args.settings is not None and Path(shared.args.settings).exists(): if shared.args.settings is not None and Path(shared.args.settings).exists():
@ -196,7 +197,7 @@ def download_model_wrapper(repo_id):
yield("Checking previously downloaded files") yield("Checking previously downloaded files")
downloader.check_model_files(model, branch, links, sha256, output_folder) downloader.check_model_files(model, branch, links, sha256, output_folder)
else: else:
yield("Downloading files") yield(f"Downloading files to {output_folder}")
downloader.download_model_files(model, branch, links, sha256, output_folder, threads=1) downloader.download_model_files(model, branch, links, sha256, output_folder, threads=1)
yield("Done!") yield("Done!")
except: except:
@ -220,7 +221,7 @@ def create_model_menus():
shared.gradio['custom_model_menu'] = gr.Textbox(label="Download Custom Model", shared.gradio['custom_model_menu'] = gr.Textbox(label="Download Custom Model",
info="Enter Hugging Face username/model path e.g: facebook/galactica-125m") info="Enter Hugging Face username/model path e.g: facebook/galactica-125m")
with gr.Column(): with gr.Column():
shared.gradio['download_button'] = gr.Button("Download", show_progress=True) shared.gradio['download_button'] = gr.Button("Download")
shared.gradio['download_status'] = gr.Markdown() shared.gradio['download_status'] = gr.Markdown()
with gr.Column(): with gr.Column():
pass pass