train能否支持baichuan，qwen等其他模型

[wzwz180]

/DiverseEvol/configs/config_example_kc_dolly.yml
的设置中model_name_or_path: decapoda-research/llama-7b-hf 如果换成其他模型的路径会报各种各样的错
train是否只支持llama模型的训练？

[danielwusg]

感谢你的问题！

理论上可以支持其他模型，只需要将tokenizer和model相关的代码 (utils.py) 替换成对应模型的方法即可；另外在config.yml中也需要根据模型内部模块指定参数 fsdp_transformer_layer_cls_to_wrap。