fix(tortoise): do not set device, which is handled in BaseTrainerModel

[ci skip]
boldsbrainai · Jan 16, 2025 · 008912c · 008912c
1 parent ce9f1bf
commit 008912c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/TTS/tts/models/tortoise.py b/TTS/tts/models/tortoise.py
@@ -342,7 +342,6 @@ def __init__(self, config: Coqpit):
             else self.args.autoregressive_batch_size
         )
         self.enable_redaction = self.args.enable_redaction
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         if self.enable_redaction:
             self.aligner = Wav2VecAlignment()