Upload 2 files

__init__.py

@@ -0,0 +1,15 @@
+from .config import *
+from .model import *
+from .tokenizer import *
+
+
+def check_install(cuda: bool = False):
+    import torch
+
+    from .version import VERSION
+
+    if cuda:
+        assert torch.cuda.is_available(), "CUDA is not available!"
+        print("CUDA available")
+
+    print(f"OLMo v{VERSION} installed")

aliases.py

@@ -0,0 +1,7 @@
+from os import PathLike
+from typing import Union
+
+__all__ = ["PathOrStr"]
+
+
+PathOrStr = Union[str, PathLike]