From 9e2b00e1678476d6aa71d27c640a93670b2b7b5e Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Wed, 7 Feb 2024 03:19:33 +1100
Subject: [PATCH] __version__

---
 pyproject.toml             | 2 +-
 unsloth/__init__.py        | 1 +
 unsloth/models/__init__.py | 1 +
 3 files changed, 3 insertions(+), 1 deletion(-)

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__