Update _utils.py

2024-07-08 10:01:20 -07:00 · 2024-07-08 10:01:20 -07:00 · ea330bdc73
commit ea330bdc73
parent 36488b5867
1 changed files with 1 additions and 1 deletions
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@ -400,7 +400,7 @@ def get_statistics():
        elif vram <= 40: vram = 40
        elif vram <= 48: vram = 48
        elif vram <= 80: vram = 80
-        else: vram = "80+"
+        else: vram = 96
        _get_statistics(f"vram-{vram}")
    except:
        pass