模型训练

汇聚全球AI模型训练与微调项目,LoRA、全参微调、蒸馏、训练框架工具大全。

共收录 524 个开源项目

kg-llm 开源

用大模型补全知识图谱,论文级代码直接复现

kg-llm 是一个探索将大语言模型用于知识图谱补全任务的开源项目,相关论文被 ICASSP 2025 接收。知···

★ 162 评分 2023-05-22
llm-mcts 开源

用蒙特卡洛树搜索增强LLM,让日常任务规划更靠谱

llm-mcts 是一个将大型语言模型与蒙特卡洛树搜索(MCTS)结合的框架,旨在提升日常任务规划中的推理···

★ 303 评分 2023-05-22
gorilla 开源

让大模型学会精准调用API,连接真实世界工具

Gorilla 是一个专注于训练和评估大语言模型(LLM)函数调用(工具调用)能力的开源项目。它解决的核···

★ 13044 评分 2023-05-19
paradigms 开源

并行采样加速扩散模型,少步生成高质量图像

Paradigms 是 NeurIPS 2023 Spotlight 论文《Parallel Sampling of Diffusion Models》的官方 PyTo···

★ 159 评分 2023-05-19
hands-on-train-and-deploy-ml 开源

10步搞定ML模型训练到API部署,以加密货币预测为例

这是一个端到端的机器学习项目实战教程,旨在帮助开发者用10个步骤完成从模型训练到部署的完整流程···

★ 891 评分 2023-05-16
aws-mlops-handson 开源

在AWS上快速搭建CTR预测的MLOps全流程

这是一个面向点击率(CTR)预测场景的AWS云上MLOps实战项目,提供了一套完整的机器学习基础设施。它···

★ 63 评分 2023-05-13
safetensors_util 开源

轻松管理 Safetensors 模型,玩转 Stable Diffusion 与 LoRA

safetensors_util 是一个用于处理 Safetensors 文件的 Python 工具库,主要面向 Stable Diffusion ···

★ 102 评分 2023-05-11
EasyEdit 开源

像改数据库一样,直接修改大模型的知识,无需重训练。

EasyEdit 是一个面向大语言模型(LLM)的知识编辑框架,由 ACL 2024 论文提出,旨在高效、精准地修···

★ 2928 评分 2023-05-09
Chattiori-Model-Merger 开源

按比例混合扩散模型,快速生成自定义风格模型

Chattiori-Model-Merger 是一个用于合并潜在扩散模型(如 Stable Diffusion)的命令行工具,支持按···

★ 53 评分 2023-05-05
lora-prompt-tool 开源

管理 LoRA 触发词,自动补全提示词,提升出图效率。

lora-prompt-tool 是一个 Stable Diffusion WebUI 扩展,用于管理 LoRA 或其他模型的触发词。它解决···

★ 257 评分 2023-05-04
litgpt 开源

一条命令搞定大模型预训练、微调和部署,省时省力

LitGPT 是一个基于 PyTorch 的高性能大语言模型(LLM)训练与部署框架,由 Lightning AI 团队开发。···

★ 13685 评分 2023-05-04
controlnet_stable_tensorrt 开源

用TensorRT加速Stable Diffusion+ControlNet,让可控生成更快更省显存。

这是一个将 Stable Diffusion 与 ControlNet 结合,并通过 TensorRT 进行推理加速的早期开源项目。···

★ 73 评分 2023-04-28
llm-foundry 开源

开箱即用的生产级 LLM 训练框架,快速搞定预训练与微调

llm-foundry 是 Databricks 开源的 LLM 训练框架,旨在为大规模基础模型训练提供生产级代码库。它解···

★ 4449 评分 2023-04-28
LLMtuner 开源

几行代码微调大模型,文本语音全覆盖

LLMtuner 是一个旨在简化大语言模型微调过程的 Python 工具库,口号是“几行代码微调大模型”。它支···

★ 252 评分 2023-04-28
CLIPA 开源

用更少数据和算力,训练出媲美大模型的CLIP

CLIPA是NeurIPS 2023论文的官方开源实现,核心发现是CLIP训练中存在逆缩放定律,即模型容量与数据规···

★ 323 评分 2023-04-26
h2o-llmstudio 开源

不用写代码,拖拽配置就能微调大模型

H2O LLM Studio 是一个用于大语言模型(LLM)微调的开源框架和无代码图形界面(GUI)。它解决了开发···

★ 5182 评分 2023-04-17
kohya-config-webui 开源

可视化点选,一键生成SD训练配置,告别手写TOML

这是一个为kohya_ss训练脚本生成TOML配置文件的Web界面工具。它解决的是Stable Diffusion模型训练中···

★ 92 评分 2023-04-14
axolotl 开源

一条命令微调大模型,省心省力

axolotl 是一个用于微调大型语言模型的开源工具,旨在简化从数据准备到模型训练的全流程。它解决了···

★ 12506 评分 2023-04-14
AutoGPTQ 开源

一键量化大模型,显存减半推理加速

AutoGPTQ 是一个基于 GPTQ 算法的 LLM 量化工具包,提供简洁易用的 API,帮助开发者将大语言模型压···

★ 5067 评分 2023-04-13
OneTrainer 开源

图形化界面搞定扩散模型训练,从 LoRA 到全量微调一站式完成。

OneTrainer 是一个面向扩散模型(Diffusion Model)的一站式训练工具,旨在简化图像生成模型的微调···

★ 3218 评分 2023-04-12
Firefly 开源

一键微调 20+ 主流大模型,快速定制专属 AI

Firefly 是一个开源的大模型训练工具,专注于高效微调和训练各类主流大语言模型。它解决了开发者在···

★ 6652 评分 2023-04-02
fasttrackml 开源

高速实验跟踪,让海量ML实验不卡顿

fasttrackml 是一个专注于速度和可扩展性的实验跟踪服务器,用 Go 语言编写。它解决机器学习团队在···

★ 119 评分 2023-03-30
self-refine 开源

让大模型自己挑错改错,越改越好

self-refine 是一个研究项目,提出了一种让大语言模型自我改进的通用框架。它不依赖额外训练或外部···

★ 822 评分 2023-03-28
LMFlow 开源

一键微调大模型,人人可用的高效工具包

LMFlow 是一个面向大型基础模型的高可扩展微调与推理工具包,旨在让每个人都能轻松使用和定制大模型···

★ 8487 评分 2023-03-27