Add device method to BiCodecTokenizer
Browse files
sparktts/models/audio_tokenizer.py
CHANGED
|
@@ -53,6 +53,11 @@ class BiCodecTokenizer:
|
|
| 53 |
f"{self.model_dir}/wav2vec2-large-xlsr-53"
|
| 54 |
).to(self.device)
|
| 55 |
self.feature_extractor.config.output_hidden_states = True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
def get_ref_clip(self, wav: np.ndarray) -> np.ndarray:
|
| 58 |
"""Get reference audio clip for speaker embedding."""
|
|
|
|
| 53 |
f"{self.model_dir}/wav2vec2-large-xlsr-53"
|
| 54 |
).to(self.device)
|
| 55 |
self.feature_extractor.config.output_hidden_states = True
|
| 56 |
+
|
| 57 |
+
def to(self, device: torch.device):
|
| 58 |
+
self.device = device
|
| 59 |
+
self.model.to(self.device)
|
| 60 |
+
self.feature_extractor.to(self.device)
|
| 61 |
|
| 62 |
def get_ref_clip(self, wav: np.ndarray) -> np.ndarray:
|
| 63 |
"""Get reference audio clip for speaker embedding."""
|