Flux9665 commited on
Commit
80a3f4b
1 Parent(s): 1327289

Update InferenceInterfaces/ControllableInterface.py

Browse files
InferenceInterfaces/ControllableInterface.py CHANGED
@@ -17,6 +17,7 @@ class ControllableInterface:
17
  os.environ["CUDA_VISIBLE_DEVICES"] = f"{gpu_id}"
18
  self.device = "cuda" if gpu_id != "cpu" else "cpu"
19
  self.model = ToucanTTSInterface(device=self.device, tts_model_path=None)
 
20
 
21
  def read(self,
22
  prompt,
 
17
  os.environ["CUDA_VISIBLE_DEVICES"] = f"{gpu_id}"
18
  self.device = "cuda" if gpu_id != "cpu" else "cpu"
19
  self.model = ToucanTTSInterface(device=self.device, tts_model_path=None)
20
+ self.model.set_utterance_embedding("female_high_voice.wav")
21
 
22
  def read(self,
23
  prompt,