From bbeee40ab3ee39dd16a1ebd2388bf82402776664 Mon Sep 17 00:00:00 2001
From: mrq <barry.quiggles@protonmail.com>
Date: Thu, 9 Mar 2023 00:51:13 +0000
Subject: [PATCH] forgot to convert to gigabytes

---
 tortoise/utils/device.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tortoise/utils/device.py b/tortoise/utils/device.py
index cf899e0..e4b5e6d 100755
--- a/tortoise/utils/device.py
+++ b/tortoise/utils/device.py
@@ -55,12 +55,11 @@ def get_device_vram( name=get_device_name() ):
     elif name == "cpu":
         available = psutil.virtual_memory()[4]
 
-    return available
+    return available / (1024 ** 3)
 
 def get_device_batch_size(name=None):
-    available = get_device_vram(name)
+    vram = get_device_vram(name)
 
-    vram = available / (1024 ** 3)
     # I'll need to rework this better
     # simply adding more tiers clearly is not a good way to go about it
     if vram > 14: