ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-03-03 03:39:36 +11:00
commit 245fe4716c
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/_utils.py
View file

@ -25,7 +25,7 @@ from platform import system as platform_system
platform_system = platform_system()
import math
__version__ = "2024.2"
__version__ = "2024.3"
# Get Flash Attention v2 if Ampere (RTX 30xx, A100)
major_version, minor_version = torch.cuda.get_device_capability()