Update utils.py
This commit is contained in:
parent
262f5ef18a
commit
7a03f1bec7
2
utils.py
2
utils.py
|
@ -232,7 +232,7 @@ def get_hubert_content(hmodel, wav_16k_tensor):
|
||||||
inputs = {
|
inputs = {
|
||||||
"source": feats.to(wav_16k_tensor.device),
|
"source": feats.to(wav_16k_tensor.device),
|
||||||
"padding_mask": padding_mask.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():
|
with torch.no_grad():
|
||||||
logits = hmodel.extract_features(**inputs)
|
logits = hmodel.extract_features(**inputs)
|
||||||
|
|
Loading…
Reference in New Issue