oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-20 22:13:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #7506 from Th-Underscore/fix/download-model

Browse source 
Add model download branch handling in download_model_wrapper
This commit is contained in:
oobabooga 2026-04-18 21:09:44 -03:00 • committed by GitHub
commit fe0b385dae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 21 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
download-model.py
View file

@ -57,8 +57,7 @@ class ModelDownloader:
return session
def sanitize_model_and_branch_names(self, model, branch):
if model[-1] == '/':
model = model[:-1]
model = model.removesuffix("/")
if model.startswith(base + '/'):
model = model[len(base) + 1:]

21
modules/ui_model_menu.py
View file

@ -239,8 +239,27 @@ def download_model_wrapper(repo_id, specific_file, progress=gr.Progress(), retur
downloader_module = importlib.import_module("download-model")
downloader = downloader_module.ModelDownloader()
update_queue = queue.Queue()
branch = None
try:
# Handle branch in URL
if "/tree/" in repo_id:
try:
repo_id, branch = repo_id.split("/tree/")
except Exception as e:
yield f"Error parsing branch from URL: {e}"
progress(0.0)
return
# Handle branch delimited by ":"
elif not repo_id.startswith("http") and ":" in repo_id:
try:
repo_id, branch = repo_id.split(":")
except Exception as e:
yield f"Error parsing branch from repo_id: {e}"
progress(0.0)
return
# Handle direct GGUF URLs
if repo_id.startswith("https://") and ("huggingface.co" in repo_id) and (repo_id.endswith(".gguf") or repo_id.endswith(".gguf?download=true")):
try:
@ -266,7 +285,7 @@ def download_model_wrapper(repo_id, specific_file, progress=gr.Progress(), retur
progress(0.0, "Preparing download...")
model, branch = downloader.sanitize_model_and_branch_names(repo_id, None)
model, branch = downloader.sanitize_model_and_branch_names(repo_id, branch)
yield "Getting download links from Hugging Face..."
links, sha256, is_lora, is_llamacpp, file_sizes = downloader.get_download_links_from_huggingface(model, branch, text_only=False, specific_file=specific_file)