diff --git a/extract_feature_print.py b/extract_feature_print.py index 3ab2504..605d5dd 100644 --- a/extract_feature_print.py +++ b/extract_feature_print.py @@ -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+")