Jarod/tortoise-tts
1
1
Fork 0
forked from mrq/tortoise-tts
Code Pull Requests Activity

revert muh clean code

Browse Source
This commit is contained in:
mrq 2023-03-10 00:56:29 +00:00
parent 0514f011ff
commit fec0685405
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tortoise/utils/device.py
View File

@ -61,9 +61,17 @@ def get_device_vram( name=get_device_name() ):
def get_device_batch_size(name=None): def get_device_batch_size(name=None):
vram = get_device_vram(name) vram = get_device_vram(name)
if vram > 14:
return 16
elif vram > 10:
return 8
elif vram > 7:
return 4
"""
for k, v in DEVICE_BATCH_SIZE_MAP: for k, v in DEVICE_BATCH_SIZE_MAP:
if vram > k: if vram > k:
return v return v
"""
return 1 return 1
def get_device_count(name=get_device_name()): def get_device_count(name=get_device_name()):