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

fix deepspeed config

Browse Source 
fix
This commit is contained in:
yfyeung 2025-06-18 04:44:10 +00:00
parent c571a88b59
commit 39d90356fe
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/speech_llm/ASR_LLM/whisper_llm_zh/ds_config_zero1.json
View File

@ -5,7 +5,7 @@
"loss_scale_window": 100, "loss_scale_window": 100,
"initial_scale_power": 16, "initial_scale_power": 16,
"hysteresis": 2, "hysteresis": 2,
"min_loss_scale": 0.01 "min_loss_scale": 1
}, },
"zero_optimization": { "zero_optimization": {
"stage": 1, "stage": 1,

1
egs/speech_llm/ASR_LLM/zipformer_llm_zh/model.py
View File

@ -64,7 +64,6 @@ class SPEECH_LLM(nn.Module):
self, self,
encoder_embed: nn.Module, encoder_embed: nn.Module,
encoder: EncoderInterface, encoder: EncoderInterface,
ctc_output: nn.Module,
llm: nn.Module, llm: nn.Module,
encoder_projector: nn.Module, encoder_projector: nn.Module,
): ):