Update tortoise/utils/devices.py vram issue

Added line 85 to set the name variable as it was 'None' causing vram to be incorrect
2023-04-12 05:33:30 +07:00 · 2023-04-12 05:33:30 +07:00 · eea4c68edc
parent 815ae5d707
commit eea4c68edc
1 changed files with 127 additions and 126 deletions
--- a/tortoise/utils/device.py
+++ b/tortoise/utils/device.py
@ -82,6 +82,7 @@ def get_device_vram( name=get_device_name() ):
    return available / (1024 ** 3)

 def get_device_batch_size(name=None):
+    name = get_device_name()
    vram = get_device_vram(name)

    if vram > 14: