1
1
forked from mrq/tortoise-tts

revert muh clean code

This commit is contained in:
mrq 2023-03-10 00:56:29 +00:00
parent 0514f011ff
commit fec0685405

View File

@ -61,9 +61,17 @@ def get_device_vram( name=get_device_name() ):
def get_device_batch_size(name=None):
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:
if vram > k:
return v
"""
return 1
def get_device_count(name=get_device_name()):