From 20482bf9d67ebcaa295b9c057dc71503538f4d4f Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 4 Mar 2025 02:57:54 -0800 Subject: [PATCH] __version__ --- unsloth/__init__.py | 1 + unsloth/models/__init__.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index e33d16577a..caa06b012d 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -212,6 +212,7 @@ except: pass from .models import * +from .models import __version__ from .save import * from .chat_templates import * from .tokenizer_utils import * diff --git a/unsloth/models/__init__.py b/unsloth/models/__init__.py index 29ad78dae2..e11cd54417 100644 --- a/unsloth/models/__init__.py +++ b/unsloth/models/__init__.py @@ -19,5 +19,5 @@ from .llama import FastLlamaModel from .mistral import FastMistralModel from .qwen2 import FastQwen2Model from .dpo import PatchDPOTrainer, PatchKTOTrainer -from ._utils import is_bfloat16_supported +from ._utils import is_bfloat16_supported, __version__ from .rl import PatchFastRL, vLLMSamplingParams