diff --git a/pyproject.toml b/pyproject.toml index 7f24aabbf2..3f97ec706e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,7 +37,7 @@ triton = [ ] huggingface = [ - "unsloth_zoo>=2025.3.17", + "unsloth_zoo>=2025.4.1", "packaging", "tyro", "transformers>=4.46.1,!=4.47.0", diff --git a/unsloth/__init__.py b/unsloth/__init__.py index d401b7205f..c326bfe9cb 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -198,19 +198,20 @@ pass # Check for unsloth_zoo try: unsloth_zoo_version = importlib_version("unsloth_zoo") - if Version(unsloth_zoo_version) < Version("2025.3.17"): - print( - "Unsloth: Updating Unsloth-Zoo utilies to the latest version.\n"\ - "To disable this, set `os.environ['UNSLOTH_DISABLE_AUTO_UPDATES'] = '1'`" - ) - if os.environ.get("UNSLOTH_DISABLE_AUTO_UPDATES", "0") == "0": - try: - os.system("pip install --upgrade --no-cache-dir --no-deps unsloth_zoo") - except: - try: - os.system("pip install --upgrade --no-cache-dir --no-deps --user unsloth_zoo") - except: - raise ImportError("Unsloth: Please update unsloth_zoo via `pip install --upgrade --no-cache-dir --no-deps unsloth_zoo`") + if Version(unsloth_zoo_version) < Version("2025.4.1"): + pass + # print( + # "Unsloth: Updating Unsloth-Zoo utilies to the latest version.\n"\ + # "To disable this, set `os.environ['UNSLOTH_DISABLE_AUTO_UPDATES'] = '1'`" + # ) + # if os.environ.get("UNSLOTH_DISABLE_AUTO_UPDATES", "0") == "0": + # try: + # os.system("pip install --upgrade --no-cache-dir --no-deps unsloth_zoo") + # except: + # try: + # os.system("pip install --upgrade --no-cache-dir --no-deps --user unsloth_zoo") + # except: + # raise ImportError("Unsloth: Please update unsloth_zoo via `pip install --upgrade --no-cache-dir --no-deps unsloth_zoo`") import unsloth_zoo except: raise ImportError("Unsloth: Please install unsloth_zoo via `pip install unsloth_zoo`") diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 840c15c003..e18d024522 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -__version__ = "2025.3.19" +__version__ = "2025.4.1" __all__ = [ "SUPPORTS_BFLOAT16",