Add support for return_dict_in_generate

#4
by Lockee - opened

It seems that return_dict_in_generate support is removed in CpmBeeForCausalLM, please add it back

Sign up or log in to comment