mrq/tortoise-tts
4
6
Fork 15
Code Issues 9 Pull Requests 1 Packages Projects Releases Wiki Activity

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

Browse Source 
Reviewed-on: #44
This commit is contained in:
mrq 2023-04-12 19:58:02 +00:00
commit f025470d60
1 changed files with 127 additions and 126 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tortoise/utils/device.py
View File

@ -82,6 +82,7 @@ def get_device_vram( name=get_device_name() ):
return available / (1024 ** 3) return available / (1024 ** 3)
def get_device_batch_size(name=None): def get_device_batch_size(name=None):
name = get_device_name()
vram = get_device_vram(name) vram = get_device_vram(name)
if vram > 14: if vram > 14: