diff --git a/pyproject.toml b/pyproject.toml index 5bbcc3560d..ddd8ae2962 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -23,7 +23,7 @@ classifiers = [ ] [tool.setuptools.dynamic] -version = {attr = "unsloth.models._utils.__version__"} +version = {attr = "unsloth.__version__"} [tool.setuptools] include-package-data = false diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 66c10e28e7..ec6b36e34a 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -81,4 +81,5 @@ except: pass from .models import * +from .models import __version__ from .save import * diff --git a/unsloth/models/__init__.py b/unsloth/models/__init__.py index 891947d69e..2f9285d347 100644 --- a/unsloth/models/__init__.py +++ b/unsloth/models/__init__.py @@ -16,3 +16,4 @@ from .loader import FastLanguageModel from .llama import FastLlamaModel from .mistral import FastMistralModel from .dpo import PatchDPOTrainer +from ._utils import __version__