From 51d6797c7cb24cb3d3c1d7d68d67b8bfb34dfad7 Mon Sep 17 00:00:00 2001 From: shimmy <107991372+shimmyshimmer@users.noreply.github.com> Date: Mon, 18 Dec 2023 11:32:46 +1100 Subject: [PATCH] Update __init__.py Fixing typo extra '=' --- unsloth/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index e39afddfa..5b0b8d325 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -48,7 +48,7 @@ except: # We only support torch 2.1 # Fixes https://github.com/unslothai/unsloth/issues/38 -torch_version = = torch.__version__.split(".") +torch_version = torch.__version__.split(".") major_torch, minor_torch = torch_version[0], torch_version[1] major_torch, minor_torch = int(major_torch), int(minor_torch) if (major_torch != 2) or (major_torch == 2 and minor_torch < 1):