neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-03-15 09:53:51 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove var

Browse source
This commit is contained in:
manmay-nakhashi 2023-07-16 01:06:19 +05:30
parent 19f5250454
commit be7c5056f7
1 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tortoise/api.py
View file

@ -252,12 +252,9 @@ class TextToSpeech:
self.rlg_diffusion = None
@contextmanager
def temporary_cuda(self, model):
if self.high_vram:
yield model
else:
m = model.to(self.device)
yield m
m = model.cpu()
m = model.to(self.device)
yield m
m = model.cpu()
def load_cvvp(self):
"""Load CVVP model."""