Error in computing snr: AttributeError: 'CustomPyanNetModel' object has no attribute 'example_output'

Hi! Thanks for the great work!

I run the following command to extract features:
```bash
CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --dataset_name /path/to/my/local/dataset --output_dir /path/to/my/feats/ --cpu_num_workers 32 --num_workers_per_gpu_for_pitch 8 --num_workers_per_gpu_for_snr 8
```
It seems that it successfully extracted the pitch, but encountered the following error when computing snr:

```
Model was trained with pyannote.audio 0.0.1, yours is 3.2.0. Bad things might happen unless you revert pyannote.audio to 0.x.
Model was trained with torch 1.12.1+cu102, yours is 2.3.0+cu121. Bad things might happen unless you revert torch to 1.x.
Map (num_proc=32): 0%| | 0/8000 [00:25
snr_dataset = dataset.map(
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/datasets/dataset_dict.py", line 869, in map
{
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/datasets/dataset_dict.py", line 870, in
k: dataset.map(
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/datasets/arrow_dataset.py", line 602, in wrapper
out: Union["Dataset", "DatasetDict"] = func(self, *args, **kwargs)
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/datasets/arrow_dataset.py", line 567, in wrapper
out: Union["Dataset", "DatasetDict"] = func(self, *args, **kwargs)
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/datasets/arrow_dataset.py", line 3248, in map
for rank, done, content in iflatmap_unordered(
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/datasets/utils/py_utils.py", line 718, in iflatmap_unordered
[async_result.get(timeout=0.05) for async_result in async_results]
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/datasets/utils/py_utils.py", line 718, in
[async_result.get(timeout=0.05) for async_result in async_results]
File "/root/miniconda3/envs/ds-env/lib/python3.8/site-packages/multiprocess/pool.py", line 771, in get
raise self._value
AttributeError: 'CustomPyanNetModel' object has no attribute 'example_output'
```

Could you possibly help me understand what the ```CustomPyanNetModel``` is used for and the possible reason for this issue? Thanks in advance!