This commit is contained in:
unknown
2025-12-22 18:52:37 -08:00
parent f8b9ff7a65
commit 35a1abc409
6 changed files with 19 additions and 16 deletions

View File

@@ -21,9 +21,9 @@ class Bark(TTSUtils, TTSRegistry, name='bark'):
self.vtt_path = os.path.join(self.session['process_dir'],Path(self.session['final_name']).stem+'.vtt')
using_gpu = self.session['device'] != devices['CPU']['proc']
enough_vram = self.session['free_vram_gb'] > 4.0
seed = 123456
random.seed(seed)
np.random.seed(seed)
seed = 0
#random.seed(seed)
#np.random.seed(seed)
torch.manual_seed(seed)
has_cuda = (torch.version.cuda is not None and torch.cuda.is_available())
if has_cuda:

View File

@@ -21,9 +21,9 @@ class Fairseq(TTSUtils, TTSRegistry, name='fairseq'):
self.vtt_path = os.path.join(self.session['process_dir'],Path(self.session['final_name']).stem+'.vtt')
using_gpu = self.session['device'] != devices['CPU']['proc']
enough_vram = self.session['free_vram_gb'] > 4.0
seed = 123456
random.seed(seed)
np.random.seed(seed)
seed = 0
#random.seed(seed)
#np.random.seed(seed)
torch.manual_seed(seed)
has_cuda = (torch.version.cuda is not None and torch.cuda.is_available())
if has_cuda:

View File

@@ -21,9 +21,9 @@ class Tacotron2(TTSUtils, TTSRegistry, name='tacotron'):
self.vtt_path = os.path.join(self.session['process_dir'],Path(self.session['final_name']).stem+'.vtt')
using_gpu = self.session['device'] != devices['CPU']['proc']
enough_vram = self.session['free_vram_gb'] > 4.0
seed = 123456
random.seed(seed)
np.random.seed(seed)
seed = 0
#random.seed(seed)
#np.random.seed(seed)
torch.manual_seed(seed)
has_cuda = (torch.version.cuda is not None and torch.cuda.is_available())
if has_cuda:

View File

@@ -21,9 +21,9 @@ class Vits(TTSUtils, TTSRegistry, name='vits'):
self.vtt_path = os.path.join(self.session['process_dir'],Path(self.session['final_name']).stem+'.vtt')
using_gpu = self.session['device'] != devices['CPU']['proc']
enough_vram = self.session['free_vram_gb'] > 4.0
seed = 123456
random.seed(seed)
np.random.seed(seed)
seed = 0
#random.seed(seed)
#np.random.seed(seed)
torch.manual_seed(seed)
has_cuda = (torch.version.cuda is not None and torch.cuda.is_available())
if has_cuda:

View File

@@ -21,7 +21,10 @@ class XTTSv2(TTSUtils, TTSRegistry, name='xtts'):
self.vtt_path = os.path.join(self.session['process_dir'],Path(self.session['final_name']).stem+'.vtt')
using_gpu = self.session['device'] != devices['CPU']['proc']
enough_vram = self.session['free_vram_gb'] > 4.0
torch.manual_seed(0)
seed = 0
#random.seed(seed)
#np.random.seed(seed)
torch.manual_seed(seed)
has_cuda = (torch.version.cuda is not None and torch.cuda.is_available())
if has_cuda:
self._apply_cuda_policy(using_gpu=using_gpu, enough_vram=enough_vram, seed=seed)

View File

@@ -21,9 +21,9 @@ class YourTTS(TTSUtils, TTSRegistry, name='yourtts'):
self.vtt_path = os.path.join(self.session['process_dir'],Path(self.session['final_name']).stem+'.vtt')
using_gpu = self.session['device'] != devices['CPU']['proc']
enough_vram = self.session['free_vram_gb'] > 4.0
seed = 123456
random.seed(seed)
np.random.seed(seed)
seed = 0
#random.seed(seed)
#np.random.seed(seed)
torch.manual_seed(seed)
has_cuda = (torch.version.cuda is not None and torch.cuda.is_available())
if has_cuda: