diff --git a/unsloth/__init__.py b/unsloth/__init__.py index e39afddfa9..5b0b8d325d 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):