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

Merge pull request #36 from e0xextazy/main

Browse Source 
Optimizing graphics card memory
This commit is contained in:
James Betker 2022-05-11 21:46:16 -06:00 committed by GitHub
commit 5c60c5d4f2
1 changed files with 22 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tortoise/api.py
View File

@ -225,6 +225,7 @@ class TextToSpeech:
properties. properties.
:param voice_samples: List of 2 or more ~10 second reference clips, which should be torch tensors containing 22.05kHz waveform data. :param voice_samples: List of 2 or more ~10 second reference clips, which should be torch tensors containing 22.05kHz waveform data.
""" """
with torch.no_grad():
voice_samples = [v.to('cuda') for v in voice_samples] voice_samples = [v.to('cuda') for v in voice_samples]
auto_conds = [] auto_conds = []