Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Minor bug fix

Browse Source
This commit is contained in:
oobabooga 2023-05-17 11:16:37 -03:00
parent 1a8151a2b6
commit fb91c07191
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/AutoGPTQ_loader.py
View File

@ -16,7 +16,7 @@ def load_quantized(model_name):
found_pts = list(path_to_model.glob("*.pt"))
found_safetensors = list(path_to_model.glob("*.safetensors"))
if len(found_safetensors) > 0:
if len(found_pts) > 1:
if len(found_safetensors) > 1:
logging.warning('More than one .safetensors model has been found. The last one will be selected. It could be wrong.')
use_safetensors = True

2
modules/GPTQ_loader.py
View File

@ -133,7 +133,7 @@ def find_quantized_model_file(model_name):
pt_path = found_pts[-1]
elif len(found_safetensors) > 0:
if len(found_pts) > 1:
if len(found_safetensors) > 1:
logging.warning('More than one .safetensors model has been found. The last one will be selected. It could be wrong.')
pt_path = found_safetensors[-1]