Import error

Author: carmocca

lit_stablelm/__init__.py

@@ -1,2 +1,10 @@
 from lit_stablelm.model import StableLMConfig, StableLM, build_rope_cache, apply_rope
 from lit_stablelm.tokenizer import Tokenizer
+
+from lightning_utilities.core.imports import RequirementCache
+
+if not bool(RequirementCache("torch>=2.1.0dev")):
+    raise ImportError(
+        "Lit-StableLM requires torch nightly (future torch 2.1). Please follow the installation instructions in the"
+        " repository README.md"
+    )