There was never a bug (shhh)

arcee-ai · Dec 30, 2023 · 7e9af2d · 7e9af2d
1 parent 58d9b97
commit 7e9af2d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mergekit/common.py b/mergekit/common.py
@@ -113,6 +113,9 @@ def __str__(self) -> str:
 
 
 def dtype_from_name(name: Optional[str]) -> torch.dtype:
+    if name.startswith("torch."):
+        name = name.removeprefix("torch.")
+
     if name == "bfloat16":
         return torch.bfloat16
     elif name == "float16":