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

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')
# 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):
if classifications[i] in ['pytorch', 'pt']:
if classifications[i] in ['pytorch', 'pt', 'gguf']:
links.pop(i)
# For GGUF, try to download only the Q4_K_M if no specific file is specified.