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

adding a parameter check for device.

Browse source
This commit is contained in:
dsenanayake 2024-05-13 12:57:36 +10:00
parent 572bdf3d24
commit b6822c725d
2 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tortoise/api.py
View file

@ -194,7 +194,11 @@ class TextToSpeech:
self.models_dir = models_dir
self.autoregressive_batch_size = pick_best_batch_size_for_gpu() if autoregressive_batch_size is None else autoregressive_batch_size
self.enable_redaction = enable_redaction
self.device = torch.device('cuda' if torch.cuda.is_available() else'cpu')
if device is None:
self.device = torch.device('cuda' if torch.cuda.is_available() else'cpu')
else:
self.device = torch.device(device)
if torch.backends.mps.is_available():
self.device = torch.device('mps')
if self.enable_redaction:

6
tortoise/api_fast.py
View file

@ -193,7 +193,11 @@ class TextToSpeech:
self.models_dir = models_dir
self.autoregressive_batch_size = pick_best_batch_size_for_gpu() if autoregressive_batch_size is None else autoregressive_batch_size
self.enable_redaction = enable_redaction
self.device = torch.device('cuda' if torch.cuda.is_available() else'cpu')
if device is None:
self.device = torch.device('cuda' if torch.cuda.is_available() else'cpu')
else:
self.device = torch.device(device)
if torch.backends.mps.is_available():
self.device = torch.device('mps')
if self.enable_redaction: