Flux9665 commited on
Commit
4e10cff
·
1 Parent(s): 5ad4ca1

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,