KeyError: "'_name'" Speaker Diarization

[taimoorkhokhar]

• I downloaded the models from the link inside the speaker_diarization directory

• After adding the model inside the path I ran the command

python diarization.py --wav_path tmp/mix_0000496.wav --model_init models/avg-4.pth

• I am getting the following error on console.

  File "/UniSpeech/downstreams/speaker_diarization/models/utils.py", line 32, in load_model
    model = task.build_model(cfg.model)
  File "//UniSpeech/downstreams/speaker_diarization/fairseq/fairseq/tasks/fairseq_task.py", line 335, in build_model
    model = models.build_model(cfg, self, from_checkpoint)
  File "/UniSpeech/downstreams/speaker_diarization/fairseq/fairseq/models/__init__.py", line 102, in build_model
    f"Could not infer model type from {cfg}. "
KeyError: "'_name'"

[taimoorkhokhar]

Solved! anyone facing the same issue go ahead and install fairseq from UniSpeech/src folder!