transformers pytorch baichuan text-generation custom_code dataset:yuekai/belle_platypus_shargpt4 dataset:yuekai/train_3.5M_CN_sampling_category_vicuna autotrain_compatible text-generation-inference endpoints_compatible region:us