Fix torch_dtype check.

explosion · Nov 13, 2023 · 05cffab · 05cffab
1 parent 9f3cca3
commit 05cffab
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/spacy_llm/models/hf/base.py b/spacy_llm/models/hf/base.py
@@ -53,7 +53,11 @@ def __init__(
                     self._config_init.pop("device")
 
         # Fetch proper torch.dtype, if specified.
-        if has_torch and self._config_init.get("torch_dtype", "") not in ("", "auto"):
+        if (
+            has_torch
+            and "torch_dtype" in self._config_init
+            and self._config_init["torch_dtype"] != "auto"
+        ):
             try:
                 self._config_init["torch_dtype"] = getattr(
                     torch, self._config_init["torch_dtype"]