From 8e7f041f121e3a36acec89b1eb58b52ccdb3f7c0 Mon Sep 17 00:00:00 2001 From: F1xx Date: Mon, 10 Nov 2025 17:40:01 +0100 Subject: [PATCH] fix(enhance): ensure audio tensors on CUDA are moved to CPU before numpy conversion --- DeepFilterNet/df/enhance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/DeepFilterNet/df/enhance.py b/DeepFilterNet/df/enhance.py index 43a7c6285..6938d5e53 100644 --- a/DeepFilterNet/df/enhance.py +++ b/DeepFilterNet/df/enhance.py @@ -188,7 +188,7 @@ def init_df( def df_features(audio: Tensor, df: DF, nb_df: int, device=None) -> Tuple[Tensor, Tensor, Tensor]: - spec = df.analysis(audio.numpy()) # [C, Tf] -> [C, Tf, F] + spec = df.analysis(audio.detach().cpu().numpy()) # [C, Tf] -> [C, Tf, F] a = get_norm_alpha(False) erb_fb = df.erb_widths() with warnings.catch_warnings():