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

forgot to convert to gigabytes

Browse Source
This commit is contained in:
mrq 2023-03-09 00:51:13 +00:00
parent 6410df569b
commit bbeee40ab3
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tortoise/utils/device.py
View File

@ -55,12 +55,11 @@ def get_device_vram( name=get_device_name() ):
elif name == "cpu": elif name == "cpu":
available = psutil.virtual_memory()[4] available = psutil.virtual_memory()[4]
return available return available / (1024 ** 3)
def get_device_batch_size(name=None): def get_device_batch_size(name=None):
available = get_device_vram(name) vram = get_device_vram(name)
vram = available / (1024 ** 3)
# I'll need to rework this better # I'll need to rework this better
# simply adding more tiers clearly is not a good way to go about it # simply adding more tiers clearly is not a good way to go about it
if vram > 14: if vram > 14: