ziyuwan / tsllm_lean

MIT License
1 stars 0 forks source link

Add prompt formatter for causal SFT trainer #1

Closed ziyuwan closed 7 months ago

ziyuwan commented 7 months ago

add a prompt formatter for the query and response building for casual sft trainer's datamodule datamodule, line 109, line 112, line 67.

Write it in a better way, e.g. create a PROMPT_FORMAT_STR and use PROMPT_FORMAT_STR.format(query=xxx, response=xxx), also QUERY_FORMAT_STR ...

ziyuwan commented 7 months ago

Please start a new branch from causal-model, e.g. by just clicking Create a branch on the right side.

ziyuwan commented 7 months ago

close since finish this feature