Merge pull request 'Update tortoise/utils/devices.py vram issue' (#44) from aJoe/tortoise-tts:main into main

Reviewed-on: #44
2023-04-12 19:58:02 +00:00 · 2023-04-12 19:58:02 +00:00 · f025470d60
commit f025470d60
parent 815ae5d707 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: