diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index cecacc0501..9eaded31a1 100644 --- 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