Skip to content

Commit 7a031d2

Browse files
committed
Rename from_pyannote to from_pretrained in SpeakerEmbedding
1 parent c5a1bae commit 7a031d2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diart/blocks/embedding.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,12 @@ def __init__(self, model: EmbeddingModel, device: Optional[torch.device] = None)
2020
self.weights_formatter = TemporalFeatureFormatter()
2121

2222
@staticmethod
23-
def from_pyannote(
23+
def from_pretrained(
2424
model,
2525
use_hf_token: Union[Text, bool, None] = True,
2626
device: Optional[torch.device] = None,
2727
) -> "SpeakerEmbedding":
28-
emb_model = EmbeddingModel.from_pyannote(model, use_hf_token)
28+
emb_model = EmbeddingModel.from_pretrained(model, use_hf_token)
2929
return SpeakerEmbedding(emb_model, device)
3030

3131
def __call__(

0 commit comments

Comments
 (0)