chore(format): run black on dev (#1697)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
parent
c2ab680e18
commit
322ce9b940
@ -116,7 +116,9 @@ else:
|
|||||||
feats = readwave(wav_path, normalize=saved_cfg.task.normalize)
|
feats = readwave(wav_path, normalize=saved_cfg.task.normalize)
|
||||||
padding_mask = torch.BoolTensor(feats.shape).fill_(False)
|
padding_mask = torch.BoolTensor(feats.shape).fill_(False)
|
||||||
inputs = {
|
inputs = {
|
||||||
"source": feats.half().to(device) if is_half and device not in ["mps", "cpu"] else feats.to(device),
|
"source": feats.half().to(device)
|
||||||
|
if is_half and device not in ["mps", "cpu"]
|
||||||
|
else feats.to(device),
|
||||||
"padding_mask": padding_mask.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, # layer 9
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user