Rename from_pyannote to from_pretrained in SpeakerEmbedding

juanmc2005 · Nov 19, 2023 · 9e894cf · 9e894cf
1 parent a71ec53
commit 9e894cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/diart/blocks/embedding.py b/src/diart/blocks/embedding.py
@@ -20,12 +20,12 @@ def __init__(self, model: EmbeddingModel, device: Optional[torch.device] = None)
         self.weights_formatter = TemporalFeatureFormatter()
 
     @staticmethod
-    def from_pyannote(
+    def from_pretrained(
         model,
         use_hf_token: Union[Text, bool, None] = True,
         device: Optional[torch.device] = None,
     ) -> "SpeakerEmbedding":
-        emb_model = EmbeddingModel.from_pyannote(model, use_hf_token)
+        emb_model = EmbeddingModel.from_pretrained(model, use_hf_token)
         return SpeakerEmbedding(emb_model, device)
 
     def __call__(