mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 09:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

add missing comma

Browse Source
This commit is contained in:
Teo 2025-07-10 16:21:42 +09:00
parent 2461a537dd
commit e7d3c003ec
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/speech_llm/ASR_LLM/whisper_llm_zh/decode.py
View File

@ -548,7 +548,7 @@ def main():
# torch.save(avg_checkpoint, filename)
else:
checkpoint = torch.load(
f"{params.exp_dir}/epoch-{params.epoch}/pytorch_model.bin", weights_only=False
f"{params.exp_dir}/epoch-{params.epoch}/pytorch_model.bin", weights_only=False,
map_location="cpu",
)
model.load_state_dict(checkpoint, strict=False)