Update extract_feature_print.py

extract_feature_print.py

@@ -1,4 +1,13 @@
 import os, sys, traceback
+from transformers import HubertModel
+import os
+from torch import nn
+
+#HuggingFacePlaceHolder = None
+class HubertModelWithFinalProj(HubertModel):
+    def __init__(self, config):
+        super().__init__(config)
+        self.final_proj = nn.Linear(config.hidden_size, config.classifier_proj_size)
 
 os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
@@ -60,7 +69,9 @@ def readwave(wav_path, normalize=False):
     feats = feats.view(1, -1)
     return feats
 
-
+if os.path.split(model_path)[0] == "/kaggle/working/Mangio-RVC-Fork/Custom" == False:
+    model_path = "/kaggle/working/Mangio-RVC-Fork/hubert_base.pt"
+    Custom_Embed = True
 # HuBERT model
 printt("load model(s) from {}".format(model_path))
 # if hubert model is exist
@@ -74,7 +85,7 @@ models, saved_cfg, task = checkpoint_utils.load_model_ensemble_and_task(
     [model_path],
     suffix="",
 )
-if os.path.split(model_path)[0] == "/kaggle/working/Mangio-RVC-Fork/Custom" == False:
+if Custom_Embed == False:
     model = models[0]
 else:
     model = HubertModelWithFinalProj.from_pretrained(model_path)