Generation config needed to be fixed

#13
by smart-liu - opened

In the generation_config.py, the "eos_token_id" is set to 2, but it should be the index of <|im_end|> token. If use other framework to deploy this model, the generation wounld stop.

InternLM org
ZwwWayne changed discussion status to closed
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment