模型训练

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

共收录 626 个开源项目

kohya_ss-docker 开源 免费 仅开源

在 Linux 上用 Docker 一键跑 kohya_ss,轻松微调 SD 模型

kohya_ss-docker 是一个将 kohya_ss 训练 WebUI 移植到 Linux 的 Docker 容器化项目。它解决了 koh···

★ 116 2023-01-28
kohya-sd-scripts-webui 开源 免费 仅开源

把 SD 模型训练变成点鼠标,不用记命令

这是一个为 kohya 的 sd-scripts 训练工具链打造的 Gradio 图形界面封装,主要解决 Stable Diffusi···

★ 169 2023-01-26
physicsnemo 开源 免费 可打开

用物理规律训练神经网络,让仿真又快又准

PhysicsNeMo(原 NVIDIA Modulus)是 NVIDIA 开源的物理机器学习(Physics-ML)深度学习框架,基于···

★ 3432 入门 · 约 10 分钟 · 库/依赖 2023-01-26
lora-scripts 开源 免费 仅开源

傻瓜式训练 Stable Diffusion 模型,点点鼠标就能炼出专属 LoRA

lora-scripts 是一个基于 kohya-ss 训练器的 Stable Diffusion 模型训练工具,提供 LoRA 和 Dreamb···

★ 6225 入门 · 约 15 分钟 · 本地安装 2023-01-22
dreambooth-keras 开源 免费 仅开源

用 TensorFlow 微调 Stable Diffusion,实现个性化图像生成。

dreambooth-keras 是一个基于 KerasCV 和 TensorFlow 的 DreamBooth 实现,旨在让开发者无需依赖 P···

★ 88 2023-01-15
sd-leap-booster 开源 免费 仅开源

让 Stable Diffusion 微调更快,从局部最优起步,省时省力。

sd-leap-booster 是一个面向 Stable Diffusion 的微调加速工具,通过引入一个“助推器”模型,将初···

★ 112 2023-01-07
stable-diffusion-keras-ft 开源 免费 仅开源

用Keras轻松微调Stable Diffusion,定制专属图像生成模型

这是一个基于Keras和TensorFlow的Stable Diffusion微调工具,旨在让开发者无需深入理解扩散模型内部···

★ 64 2022-12-24
redco 开源 免费 仅开源

一键自动化分布式训练推理,轻松搞定联邦学习与隐私保护

RedCoast是一个轻量级工具,旨在自动化分布式训练和推理流程,特别针对大语言模型和扩散模型。它解···

★ 69 2022-12-19
lora 开源 免费 仅开源

用少量图片和普通显卡,快速训练专属图像模型

LoRA(低秩适应)是一种高效的模型微调技术,最初由微软提出,此项目将其应用于扩散模型(如Stable···

★ 7557 进阶 · 约 10 分钟 · 库/依赖 2022-12-08
peft 开源 免费 仅开源

用极低成本微调大模型,省显存省时间,效果不打折。

PEFT 是一个专注于参数高效微调(Parameter-Efficient Fine-Tuning)的 Python 库,由 Hugging Fac···

★ 21881 入门 · 约 10 分钟 · 库/依赖 2022-11-25
EasyLM 开源 免费 仅开源

用 JAX 轻松搞定 LLM 训练、微调和部署

EasyLM 是一个基于 JAX/Flax 构建的大语言模型(LLM)一站式解决方案,旨在简化 LLM 的预训练、微调···

★ 2517 进阶 · 约 20 分钟 · 本地安装 2022-11-22
Dreambooth 开源 免费 仅开源

用几张照片微调扩散模型,让AI学会画你的专属对象

Dreambooth 是一个基于扩散模型的个性化微调项目,最初由 Google Research 提出,旨在通过少量(3-···

★ 97 2022-11-22
FLNK 开源 免费 仅开源

让联邦学习系统持续吸收新知识,延长寿命、降低成本

FLNK(Federated Learning with New Knowledge)是一个探索将新知识融入现有联邦学习系统的开源研究···

★ 88 2022-11-09
simple-diffusion 开源 免费 仅开源

用最简代码看懂扩散模型,快速跑通图像生成

simple-diffusion 是一个用 PyTorch 实现的最小化去噪扩散模型,旨在以极简代码展示扩散模型的核心···

★ 137 2022-10-20
xmtf 开源 免费 仅开源

多任务微调,让模型学会跨语言举一反三

xmtf 是一个通过多任务微调实现跨语言泛化的开源模型项目。它主要解决多语言自然语言处理中,模型在···

★ 534 入门 · 约 25 分钟 · 无需部署 2022-10-18
gflownet 开源 免费 仅开源

按奖励比例采样多样化候选解的生成模型框架

gflownet 是 Generative Flow Networks(生成流网络)的官方开源实现,用 PyTorch 编写,面向生成式···

★ 355 2022-10-17
Flipped-Learning 开源 免费 仅开源

让模型先猜指令再作答,零样本效果更强

Flipped-Learning 是一个基于 ICLR 2023 论文的开源实现,提出了一种新颖的零样本学习范式。传统方···

★ 116 2022-10-06
datablations 开源 免费 仅开源

数据不够时,教你如何最优训练大模型

datablations 是一个研究数据受限条件下语言模型扩展规律的开源项目,由 Hugging Face 的研究人员发···

★ 352 2022-10-06
Dreambooth-Stable-Diffusion 开源 免费 仅开源

用几张照片,训练专属 Stable Diffusion 模型,生成你的定制图像。

Dreambooth-Stable-Diffusion 是一个基于 Stable Diffusion 的模型微调工具,通过结合 Dreambooth ···

★ 3214 进阶 · 约 30 分钟 · 本地安装 2022-09-17
DreamBooth 开源 免费 仅开源

用几张照片,让AI学会画你的专属对象

DreamBooth 是一个基于 Stable Diffusion 的图像生成项目,实现了 DreamBooth 训练方法,允许用户通···

★ 61 2022-09-16
merge-models 开源 免费 仅开源

按比例融合两个扩散模型,快速生成新风格模型

merge-models 是一个用于合并两个潜在扩散模型的工具,允许用户按自定义比例融合模型权重。它主要解···

★ 262 2022-09-15