23 Commits

Author	SHA1	Message	Date
Yuekai Zhang	618b686166	add sampler state_dict	2024-06-14 12:08:50 +08:00
root	7db5445d1e	add unfreeze llm option	2024-06-14 12:08:50 +08:00
root	dbe85c1f12	add speech io dataset	2024-06-14 12:08:50 +08:00
root	8226b628f4	add lora for second stage training	2024-06-14 12:08:50 +08:00
root	3195a55ac7	update test set with wenetspeech test meeting	2024-06-14 12:08:50 +08:00
root	b26d3fa596	add logging	2024-06-14 12:08:50 +08:00
root	4ebccebcc0	removing debug log	2024-06-14 12:08:50 +08:00
root	271536248f	fix decoding issue and padding to longest	2024-06-14 12:08:50 +08:00
root	eb2c255e1e	remove position ids	2024-06-14 12:08:50 +08:00
root	639feab4df	update dataset with aishell 2	2024-06-14 12:08:50 +08:00
root	8afb0d647f	fix template	2024-06-14 12:08:50 +08:00
Yuekai Zhang	16f18080be	update prompt for decoding	2024-06-14 12:08:50 +08:00
Yuekai Zhang	40e4ac480c	change prompt	2024-06-14 12:08:50 +08:00
Yuekai Zhang	68b99f456f	fix debug	2024-06-14 12:08:50 +08:00
Yuekai Zhang	8bbd06112a	add decode log	2024-06-14 12:08:50 +08:00
root	412e926941	fix down sample method	2024-06-14 12:08:50 +08:00
root	796663066f	mask unrelated labels	2024-06-14 12:08:50 +08:00
Yuekai Zhang	3ac27d5ad4	fix requirements	2024-06-14 12:08:26 +08:00
Yuekai Zhang	09ec0d6553	add requirements.txt	2024-06-14 12:08:26 +08:00
Yuekai Zhang	19b5b86f9b	fix decoding issues	2024-06-14 12:08:26 +08:00
Yuekai Zhang	3dbbc29429	add decode file	2024-06-14 12:08:26 +08:00
Yuekai Zhang	b5a906cbbd	fix bugs	2024-06-14 12:08:26 +08:00
Yuekai Zhang	e495c9d732	add whisper llm	2024-06-14 12:08:26 +08:00