Small AutoGPTQ fix

2026-08-03 16:50:46 -05:00 · 2023-05-23 15:20:01 -03:00
parent c2d2ef7c13
commit fb6a00f4e5
1 changed files with 3 additions and 0 deletions
--- a/modules/AutoGPTQ_loader.py
+++ b/modules/AutoGPTQ_loader.py
@@ -20,6 +20,9 @@ def load_quantized(model_name):
                logger.warning(f'More than one {ext} model has been found. The last one will be selected. It could be wrong.')

            pt_path = found[-1]
+            if ext == '.safetensors':
+                use_safetensors = True
+
            break

    if pt_path is None: