strip model_id path

ModelCloud · Mar 1, 2025 · 7afcacf · 7afcacf
1 parent 54d90cc
commit 7afcacf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gptqmodel/models/auto.py b/gptqmodel/models/auto.py
@@ -189,6 +189,9 @@ def load(
             verify_hash: Optional[Union[str, List[str]]] = None,
             **kwargs,
     ):
+        if isinstance(model_id_or_path, str):
+            model_id_or_path = model_id_or_path.strip()
+
         # normalize config to cfg instance
         if isinstance(quantize_config, Dict):
             quantize_config = QuantizeConfig(**quantize_config)