Spaces:
Running
on
T4
Running
on
T4
try to figure out how ZeroGPU works
Browse files
InferenceInterfaces/ToucanTTSInterface.py
CHANGED
@@ -161,11 +161,11 @@ class ToucanTTSInterface(torch.nn.Module):
|
|
161 |
phones = self.text2phone.string_to_tensor(text, input_phonemes=input_is_phones).to(torch.device(self.device))
|
162 |
mel, durations, pitch, energy = self.phone2mel(phones,
|
163 |
return_duration_pitch_energy=True,
|
164 |
-
utterance_embedding=self.default_utterance_embedding,
|
165 |
durations=durations,
|
166 |
pitch=pitch,
|
167 |
energy=energy,
|
168 |
-
lang_id=self.lang_id,
|
169 |
duration_scaling_factor=duration_scaling_factor,
|
170 |
pitch_variance_scale=pitch_variance_scale,
|
171 |
energy_variance_scale=energy_variance_scale,
|
|
|
161 |
phones = self.text2phone.string_to_tensor(text, input_phonemes=input_is_phones).to(torch.device(self.device))
|
162 |
mel, durations, pitch, energy = self.phone2mel(phones,
|
163 |
return_duration_pitch_energy=True,
|
164 |
+
utterance_embedding=self.default_utterance_embedding.to(device),
|
165 |
durations=durations,
|
166 |
pitch=pitch,
|
167 |
energy=energy,
|
168 |
+
lang_id=self.lang_id.to(device),
|
169 |
duration_scaling_factor=duration_scaling_factor,
|
170 |
pitch_variance_scale=pitch_variance_scale,
|
171 |
energy_variance_scale=energy_variance_scale,
|