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

Minor fix

Browse Source
This commit is contained in:
oobabooga 2023-12-08 05:01:25 -08:00
parent 2c5a1e67f9
commit 7bbe7e803a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
download-model.py
View File

@ -128,7 +128,7 @@ class ModelDownloader:
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.
# If not present, exclude all GGUFs, as that's like a repository with both # If not present, exclude all GGUFs, as that's likely a repository with both
# GGUF and fp16 files. # GGUF and fp16 files.
if has_gguf and specific_file is None: if has_gguf and specific_file is None:
has_q4km = False has_q4km = False
@ -138,7 +138,7 @@ class ModelDownloader:
if has_q4km: if has_q4km:
for i in range(len(classifications) - 1, -1, -1): for i in range(len(classifications) - 1, -1, -1):
if 'q4_k_m' in links[i].lower(): if 'q4_k_m' not in links[i].lower():
links.pop(i) links.pop(i)
else: else:
for i in range(len(classifications) - 1, -1, -1): for i in range(len(classifications) - 1, -1, -1):