forgot to convert to gigabytes

2023-03-09 00:51:13 +07:00 · 2023-03-09 00:51:13 +07:00 · bbeee40ab3
parent 6410df569b
commit bbeee40ab3
1 changed files with 2 additions and 3 deletions
--- 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: