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