ensure that torch is not used unless it is available

explosion · May 17, 2024 · 2f5047c · 2f5047c
1 parent 6fdf8c0
commit 2f5047c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spacy_llm/models/hf/stablelm.py b/spacy_llm/models/hf/stablelm.py
@@ -2,11 +2,11 @@
 
 from confection import SimpleFrozenDict
 
-from ...compat import Literal, has_transformers, torch, transformers
+from ...compat import Literal, has_torch, has_transformers, torch, transformers
 from ...registry.util import registry
 from .base import HuggingFace
 
-if has_transformers:
+if has_transformers and has_torch:
 
     class _StopOnTokens(transformers.StoppingCriteria):
         def __call__(