funasr v2 setup

游雁

2023-11-21 c644ac8f58895b9e29e9cfca79465fd2c0efaa5a

 funasr/utils/speaker_utils.py

@@ -12,7 +12,7 @@
from typing import Any, Dict, List, Union

import numpy as np
import soundfile as sf
import librosa as sf
import torch
import torchaudio
import logging
@@ -43,7 +43,7 @@
        for i in range(len(inputs)):
            if isinstance(inputs[i], str):
                file_bytes = File.read(inputs[i])
                data, fs = sf.read(io.BytesIO(file_bytes), dtype='float32')
                data, fs = sf.load(io.BytesIO(file_bytes), dtype='float32')
                if len(data.shape) == 2:
                    data = data[:, 0]
                data = torch.from_numpy(data).unsqueeze(0)