Skip to content

Commit

Permalink
Minor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
oobabooga committed Dec 8, 2023
1 parent 2c5a1e6 commit 7bbe7e8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions download-model.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,7 @@ def get_download_links_from_huggingface(self, model, branch, text_only=False, sp
links.pop(i)

# 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.
if has_gguf and specific_file is None:
has_q4km = False
Expand All @@ -138,7 +138,7 @@ def get_download_links_from_huggingface(self, model, branch, text_only=False, sp

if has_q4km:
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)
else:
for i in range(len(classifications) - 1, -1, -1):
Expand Down

0 comments on commit 7bbe7e8

Please sign in to comment.