Doesn't work for large files - Python 3.10.11 on Windows 10

Just using this simple code:

from df.enhance import enhance, init_df, load_audio, save_audio
from df.utils import download_file

if __name__ == "__main__":
    # Load default model
    model, df_state, _ = init_df()
    # Download and open some audio file. You use your audio files here
    audio_path = '1.wav'
    audio, _ = load_audio(audio_path, sr=df_state.sr())
    # Denoise the audio
    enhanced = enhance(model, df_state, audio)
    # Save for listening
    save_audio("enhanced.wav", enhanced, df_state.sr())

I can process 50 minute file easily, but 60 minute file (329 MB) causes this error after model is loaded:

2024-01-18 15:22:15 | INFO | DF | Running on device cuda:0 2024-01-18 15:22:15 | INFO | DF | Model loaded Traceback (most recent call last): File "W:\Whisper\Deep-filter\pokus1.py", line 11, in enhanced = enhance(model, df_state, audio) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\utils_contextlib.py", line 115, in decorate_context return func(*args, kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\df\enhance.py", line 235, in enhance enhanced = model(spec.clone(), erb_feat, spec_feat)[0].cpu() File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl return forward_call(args, kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\df\deepfilternet3.py", line 411, in forward e0, e1, e2, e3, emb, c0, lsnr = self.enc(feat_erb, feat_spec) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl return forward_call(*args, *kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\df\deepfilternet3.py", line 177, in forward c0 = self.df_conv0(feat_spec) # [B, C, T, Fc] File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl return self._call_impl(args, kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl return forward_call(*args, kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\container.py", line 215, in forward input = module(input) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl return forward_call(args, kwargs) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\conv.py", line 460, in forward return self._conv_forward(input, self.weight, self.bias) File "W:\Whisper\Portable\Bin\lib\site-packages\torch\nn\modules\conv.py", line 456, in _conv_forward return F.conv2d(input, weight, bias, self.stride, RuntimeError: Expected canUse32BitIndexMath(input) && canUse32BitIndexMath(output) to be true, but got false. (Could this error message be improved? If so, please report an enhancement request to PyTorch.)

Any idea?

Rikorose / DeepFilterNet

Doesn't work for large files - Python 3.10.11 on Windows 10 #497