Skip to content

Commit ad67858

Browse files
committed
fix embedding format and device
1 parent a1071b0 commit ad67858

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

identification/speaker_identify.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -353,6 +353,8 @@ def speaker_identify(
353353
break
354354

355355
embedding_audio = compute_embedding(audio_selection)
356+
embedding_audio = torch.from_numpy(embedding_audio)
357+
embedding_audio = embedding_audio.to(_embedding_model.device)
356358

357359
# Loop on the target speakers
358360
for speaker_name in speaker_names:
@@ -362,6 +364,7 @@ def speaker_identify(
362364
# Get speaker embedding
363365
with open(_get_speaker_embedding_file(speaker_name), "rb") as f:
364366
embedding_speaker = pkl.load(f)
367+
embedding_speaker = torch.from_numpy(embedding_speaker)
365368
embedding_speaker = embedding_speaker.to(_embedding_model.device)
366369

367370
# Compute score similarity

0 commit comments

Comments
 (0)