Dupaja commited on
Commit
617d2a4
·
1 Parent(s): 9cf2190

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -21,8 +21,9 @@ class EndpointHandler:
21
 
22
  with torch.device("cuda"):
23
  self.model= SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
24
- self.processor = SpeechT5Processor.from_pretrained(checkpoint)
25
- self.vocoder = SpeechT5HifiGan.from_pretrained(vocoder_id)
 
26
 
27
  embeddings_dataset = load_dataset(dataset_id, split="validation", trust_remote_code=True)
28
  self.embeddings_dataset = embeddings_dataset
 
21
 
22
  with torch.device("cuda"):
23
  self.model= SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
24
+
25
+ self.processor = SpeechT5Processor.from_pretrained(checkpoint)
26
+ self.vocoder = SpeechT5HifiGan.from_pretrained(vocoder_id)
27
 
28
  embeddings_dataset = load_dataset(dataset_id, split="validation", trust_remote_code=True)
29
  self.embeddings_dataset = embeddings_dataset