new hubert method

.gitattributes

@@ -2,3 +2,4 @@ ilariasuitewallpaper.jpg filter=lfs diff=lfs merge=lfs -text
 ilariaaisuite.png filter=lfs diff=lfs merge=lfs -text
 pretrained_models/giga330M.pth filter=lfs diff=lfs merge=lfs -text
 pretrained_models/encodec_4cb2048_giga.th filter=lfs diff=lfs merge=lfs -text
+hubert.pth filter=lfs diff=lfs merge=lfs -text

app.py

@@ -246,7 +246,7 @@ associated_links = {}
 def load_hubert():
     global hubert_model
     # Load the model
-    hubert_model = torch.load("hubert_base.pt", map_location=config.device)
+    hubert_model = torch.load("hubert_base.pth", map_location=config.device)
     
     # Prepare the model
     hubert_model = hubert_model.to(config.device)

hubert.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e579cfcfb99bfca12e392d89854f7ed722ebb08c74daa8d54b4b4165436e8f7
+size 377560373

vc_infer_pipeline.py

@@ -396,8 +396,9 @@ class VC(object):
         }
         t0 = ttime()
         with torch.no_grad():
-            logits = model.extract_features(**inputs)
-            feats = model.final_proj(logits[0]) if version == "v1" else logits[0]
+            feats = model(inputs["source"])["last_hidden_state"]
+            # logits = model.extract_features(**inputs)
+            # feats = model.final_proj(logits[0]) if version == "v1" else logits[0]
         if protect < 0.5 and pitch != None and pitchf != None:
             feats0 = feats.clone()
         if (