FlagAI大模型工具包
FlagAI 飞智是一个快速、易于使用和可扩展的大模型工具包。目标是支持在多模态的各种下游任务上训练、微调和部署大规模模型。
优点:
-
可通过 API 快速下载模型
提供 API 方便你快速下载模型,并在给定(中/英文)文本上使用这些预训练模型,在从SuperGLUE和CLUE benchmarks收集的广泛使用的数据集上对它们进行微调。
FlagAI 现已支持 30+ 主流模型,包括语言模型Aquila, 多模态模型 AltCLIP 、文生图模型 AltDiffusion、最高百亿参数的 悟道GLM,EVA-CLIP、Galactica、OPT、BERT、RoBERTa、GPT2、T5、ALM、Huggingface Transformers 等。
-
仅用十行代码即可进行并行训练
飞智由四个最流行的数据/模型并行库(PyTorch/Deepspeed/Megatron-LM/BMTrain)提供支持,它们之间实现了无缝集成。 你可以用不到十行代码来并行你的训练/测试过程。
-
提供提示学习工具包
FlagAI 提供了提示学习(prompt-learning)的工具包,用于少样本学习(few-shot learning)任务。
-
尤其擅长中文任务
FlagAI 目前支持的模型可以应用于文本分类、信息提取、问答、摘要、文本生成等任务,尤其擅长中文任务。
评论