优化代码结构

This commit is contained in:
源文雨 2023-06-24 16:36:15 +08:00
parent 46c0e9b2fe
commit ccba65151e
1 changed files with 1 additions and 4 deletions

View File

@ -19,14 +19,11 @@ import soundfile as sf
import numpy as np
from fairseq import checkpoint_utils
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
device = "cpu"
if torch.cuda.is_available():
device = "cuda"
elif torch.backends.mps.is_available():
device = "mps"
else:
device = "cpu"
f = open("%s/extract_f0_feature.log" % exp_dir, "a+")