Update extract_feature_print.py

extract_feature_print.py

@@ -242,7 +242,7 @@ models, saved_cfg, task = checkpoint_utils.load_model_ensemble_and_task(
     [model_path],
     suffix="",
 )
-if Custom_Embed == False:
+if Custom_Embed == False or sample_embedding == "hubert_large_ll60k":
     model = models[0]
     if device not in ["mps", "cpu"]:
         model = model.half()
@@ -275,7 +275,7 @@ else:
                     if device not in ["mps", "cpu"]
                     else feats.to(device),
                     "padding_mask": padding_mask.to(device),
-                    "output_layer": 9 if version == "v1" else 12,  # layer 9
+                    "output_layer": 9 if version == "v1" else 12 if sample_embedding != "hubert_large_ll60k" else 24,  # layer 9
                 }
                 with torch.no_grad():
                     if Custom_Embed == False: