python系列&deep_study系列:开源大模型微调工具 XTuner

开源大模型微调工具 XTuner




开源大模型微调工具 XTuner

支持几乎所有 GPU 上对 LLM 预训练或微调,可在 8GB GPU 上微调 7B , Llama3Mixtral-8x7BQwen 等模型, LMDeployOpenCompass 集成

支持完整从微调,到部署,评测的一整套工具链

https://github.com/InternLM/xtuner

高效:支持在几乎所有GPU上对LLMVLM进行预训练/微调。可在单个8GB GPU微调7B LLM,也可在多节点上微调超过70B的模型。自动分配高性能操作符,如FlashAttentionTriton内核,以提高训练吞吐量。兼容DeepSpeed,轻松利用各种ZeRO优化技术

灵活:支持各种LLMInternLMMixtral-8x7BLlama 2ChatGLMQwenBaichuan等)。支持VLM(LLaVA),特别是LLaVA-InternLM2-20B的性能出色。设计良好的数据管道,适应任何格式的数据集,包括开源和自定义格式。支持各种训练算法(QLoRALoRA、全参数微调),让用户选择最适合其需求的解决方案

功能齐全:支持连续预训练、指导微调和代理微调。支持与部署和服务器工具包(LMDeploy)以及大规模评估工具包(OpenCompassVLMEvalKit)的无缝集成。







Sanfor

开源大模型微调工具 XTuner

posted @ 2024-08-23 10:55  坦笑&&life  阅读(9)  评论(0编辑  收藏  举报  来源