AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

downloader: fix downloading 01-ai/Yi-1.5-34B-Chat

Browse Source
This commit is contained in:
oobabooga 2024-05-12 10:43:50 -07:00
parent 9557f49f2f
commit e225b0b995
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
download-model.py
View File

@ -138,9 +138,12 @@ class ModelDownloader:
cursor = cursor.replace(b'=', b'%3D') cursor = cursor.replace(b'=', b'%3D')
# If both pytorch and safetensors are available, download safetensors only # If both pytorch and safetensors are available, download safetensors only
if (has_pytorch or has_pt) and has_safetensors: # Also if GGUF and safetensors are available, download only safetensors
# (why do people do this?)
if (has_pytorch or has_pt or has_gguf) and has_safetensors:
has_gguf = False
for i in range(len(classifications) - 1, -1, -1): for i in range(len(classifications) - 1, -1, -1):
if classifications[i] in ['pytorch', 'pt']: if classifications[i] in ['pytorch', 'pt', 'gguf']:
links.pop(i) links.pop(i)
# For GGUF, try to download only the Q4_K_M if no specific file is specified. # For GGUF, try to download only the Q4_K_M if no specific file is specified.