Merge remote-tracking branch 'origin/main' into InternVL2_support

ZX-ModelCloud · Dec 25, 2024 · ad63498 · ad63498
2 parents 0f7cba5 + ebf169a
commit ad63498
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gptqmodel/utils/importer.py b/gptqmodel/utils/importer.py
@@ -63,7 +63,7 @@ def normalize_device_device_map(device: Optional[Union[str, torch.device]], devi
         return None
     else:
         if isinstance(device, str):
-            return DEVICE(device)
+            return normalize_device(device)
         elif isinstance(device, torch.device):
             return DEVICE(device.type)
         else: