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

Update tortoise/utils/devices.py vram issue #44

Merged
mrq merged 1 commits from aJoe/tortoise-tts:main into main 2023-04-12 19:58:04 +00:00
Conversation 2 Commits 1 Files Changed 1 +127 -126
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)
def get_device_batch_size(name=None):
name = get_device_name()
aJoe commented 2023-04-12 05:43:05 +00:00
Review
Copy Link

here

here
vram = get_device_vram(name)
if vram > 14: