diff --git a/src/axolotl/__init__.py b/src/axolotl/__init__.py index 820642bf0..4b9f11bbd 100644 --- a/src/axolotl/__init__.py +++ b/src/axolotl/__init__.py @@ -1,5 +1,11 @@ """Axolotl - Train and fine-tune large language models""" import pkgutil +from importlib.metadata import PackageNotFoundError, version __path__ = pkgutil.extend_path(__path__, __name__) # Make this a namespace package + +try: + __version__ = version("axolotl") +except PackageNotFoundError: + __version__ = "unknown"