Skip to content

Commit

Permalink
fix: black embedding.py
Browse files Browse the repository at this point in the history
... though it has nothing to do with this PR...
  • Loading branch information
hbredin committed Nov 10, 2023
1 parent dfd7f33 commit 0dd713f
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/diart/blocks/embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,9 @@ def from_pyannote(
device: Optional[torch.device] = None,
):
model = EmbeddingModel.from_pyannote(model, use_hf_token)
return OverlapAwareSpeakerEmbedding(model, gamma, beta, norm, normalize_weights, device)
return OverlapAwareSpeakerEmbedding(
model, gamma, beta, norm, normalize_weights, device
)

def __call__(
self, waveform: TemporalFeatures, segmentation: TemporalFeatures
Expand Down

0 comments on commit 0dd713f

Please sign in to comment.