torch>1.10.0 torchvision transformers>=4.27.1 mdtex2html gradio sentencepiece tensorboardX datasets cpm_kernels einops
