From 1f1211fbd66fe706c5e8d8293305e3607f18aba8 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 8 Jul 2024 10:01:20 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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