diff --git a/download-model.py b/download-model.py index 2642c405..dedd5f6c 100644 --- a/download-model.py +++ b/download-model.py @@ -73,7 +73,7 @@ class ModelDownloader: if not is_lora and fname.endswith(('adapter_config.json', 'adapter_model.bin')): is_lora = True - is_pytorch = re.match("(pytorch|adapter)_model.*\.bin", fname) + is_pytorch = re.match("(pytorch|adapter|gptq)_model.*\.bin", fname) is_safetensors = re.match(".*\.safetensors", fname) is_pt = re.match(".*\.pt", fname) is_ggml = re.match(".*ggml.*\.bin", fname)