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