mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-24 09:16:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

typo fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2023-11-09 12:44:02 +08:00
parent bd09a9eca8
commit bf83497475
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
kaldifeat/python/csrc/utils.cc
View File

@ -130,7 +130,7 @@ WhisperFbankOptions WhisperFbankOptionsFromDict(py::dict dict) {
opts.frame_opts = FrameExtractionOptionsFromDict(dict["frame_opts"]); opts.frame_opts = FrameExtractionOptionsFromDict(dict["frame_opts"]);
} }
AS_DICT(num_mels); FROM_DICT(int_, num_mels);
if (dict.contains("device")) { if (dict.contains("device")) {
opts.device = torch::Device(std::string(py::str(dict["device"]))); opts.device = torch::Device(std::string(py::str(dict["device"])));
@ -144,6 +144,8 @@ py::dict AsDict(const WhisperFbankOptions &opts) {
dict["frame_opts"] = AsDict(opts.frame_opts); dict["frame_opts"] = AsDict(opts.frame_opts);
AS_DICT(num_mels);
auto torch_device = py::module_::import("torch").attr("device"); auto torch_device = py::module_::import("torch").attr("device");
dict["device"] = torch_device(opts.device.str()); dict["device"] = torch_device(opts.device.str());