vrr/unsloth
2
0
Fork 0
mirror of https://github.com/unslothai/unsloth.git synced 2026-07-09 15:58:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 29

Update __init__.py

Browse source 
Fixing typo extra '='
This commit is contained in:
shimmy 2023-12-18 11:32:46 +11:00 • committed by Daniel Han
parent feec338c08
commit 51d6797c7c
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/__init__.py
View file

@ -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):