search

Rikorose / DeepFilterNet

Noise supression using deep filtering
https://huggingface.co/spaces/hshr/DeepFilterNet2
Other
2.15k stars 198 forks source link

Setting gpu #521

Open rizwanishaq opened 3 months ago

rizwanishaq commented 3 months ago

Is there any way that we set another gpu like cuda:1?

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

model, dfstate, = init_df()

enricguso commented 2 months ago

import os os.environ["CUDA_VISIBLE_DEVICES"]="1"