-
Notifications
You must be signed in to change notification settings - Fork 975
Open
Description
Tested versions
1.1
System information
linux
Issue description
I download the model file locally, and I run the demo, the runtime code reports an error:einops.EinopsError: Error while processing rearrange-reduction pattern "(c s) d -> c s d".
my demo is:
from pyannote.audio import Pipeline
pipeline = Pipeline.from_pretrained(
"./pyannote/",
use_auth_token="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"
)
import torch
pipeline.to(torch.device("cuda"))
print(pipeline)
run the pipeline on an audio file
diarization = pipeline("176cbf4327eef6458cc63c1e9a52fd73.wav")
print(diarization)
dump the diarization output to disk using RTTM format
with open("audio.rttm", "w") as rttm:
diarization.write_rttm(rttm)
for turn, , speaker in diarization.itertracks(yield_label=True):
print(f"start={turn.start:.1f}s stop={turn.end:.1f}s speaker{speaker}")
Minimal reproduction example (MRE)
none
Metadata
Metadata
Assignees
Labels
No labels