Update extract_feature_print.py

extract_feature_print.py

@@ -140,7 +140,7 @@ class HubertModelWithFinalProj(HubertModel):
 os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
 
-# device=sys.argv[1]
+device=sys.argv[1]
 n_part = int(sys.argv[2])
 i_part = int(sys.argv[3])
 if len(sys.argv) == 6:
@@ -157,7 +157,7 @@ import soundfile as sf
 import numpy as np
 from fairseq import checkpoint_utils
 
-device = "cpu"
+#device = "cpu"
 if torch.cuda.is_available():
     device = "cuda"
 elif torch.backends.mps.is_available():