TextGen: Implementation of Text Generation models, include LLaMA, BLOOM, GPT2, BART, T5, SongNet and so on. 文本生成模型，实现了包括LLaMA，ChatGLM，BLOOM，GPT2，Seq2Seq，BART，T5，UDA等模型的训练和预测，开箱即用。
Apache License 2.0
937
stars
109
forks
source link
Fixed bug: Handle error when CUDA is not available #52
Got a bug when running on a Mac with Apple Silicon or an AMD GPU or CUDA is not available.