Update utils.py

2023-04-14 10:54:34 +08:00 · 2023-04-14 10:54:34 +08:00 · 7a03f1bec7
parent 262f5ef18a
commit 7a03f1bec7
1 changed files with 1 additions and 1 deletions
--- a/utils.py
+++ b/utils.py
@ -232,7 +232,7 @@ def get_hubert_content(hmodel, wav_16k_tensor):
  inputs = {
    "source": feats.to(wav_16k_tensor.device),
    "padding_mask": padding_mask.to(wav_16k_tensor.device),
-    "output_layer": 9,  # layer 9
+    "output_layer": 12,  # layer 12
  }
  with torch.no_grad():
    logits = hmodel.extract_features(**inputs)