aJoe/tortoise-tts
1
0
Fork 0
forked from mrq/tortoise-tts
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update tortoise/utils/devices.py vram issue

Browse Source 
Added line 85 to set the name variable as it was 'None' causing vram to be incorrect
This commit is contained in:
aJoe 2023-04-12 05:33:30 +00:00
parent 815ae5d707
commit eea4c68edc
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: