模型训练

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

共收录 524 个开源项目

redco 开源

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

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

★ 69 评分 2022-12-19
sdweb-merge-block-weighted-gui 开源

逐块调权重,精细融合模型风格

这是一个针对 Stable Diffusion WebUI(AUTOMATIC1111 版)的扩展插件,用于精细控制模型合并时的权···

★ 320 评分 2022-12-15
lora 开源

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

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

★ 7557 评分 2022-12-08
peft 开源

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

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

★ 21727 评分 2022-11-25
EasyLM 开源

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

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

★ 2517 评分 2022-11-22
Dreambooth 开源

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

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

★ 97 评分 2022-11-22
Merge-Stable-Diffusion-models-withou··· 开源

不训练,无损合并多个Stable Diffusion模型,统一风格与能力。

这是一个将Git Re-Basin论文中提出的模型合并方法(通过排列对称性合并模型)应用于Stable Diffusi···

★ 147 评分 2022-11-16
FLNK 开源

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

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

★ 88 评分 2022-11-09
simple-diffusion 开源

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

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

★ 137 评分 2022-10-20
xmtf 开源

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

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

★ 533 评分 2022-10-18
Flipped-Learning 开源

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

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

★ 117 评分 2022-10-06
datablations 开源

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

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

★ 345 评分 2022-10-06
mlops-hf-tf-vision-models 开源

把Hugging Face视觉模型一键部署到生产流水线

这是一个专注于视觉模型MLOps的开源项目,基于Hugging Face Transformers和TensorFlow Extended (T···

★ 110 评分 2022-09-25
Dreambooth-Stable-Diffusion 开源

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

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

★ 3212 评分 2022-09-17
DreamBooth 开源

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

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

★ 61 评分 2022-09-16
merge-models 开源

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

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

★ 263 评分 2022-09-15
TimeSeriesGAN 开源

用 GAN 生成逼真时间序列,搞定数据增强和异常检测

TimeSeriesGAN 是一个基于生成对抗网络(GAN)的时间序列分析工具,专注于合成数据生成、异常检测和···

★ 70 评分 2022-09-14
lora-svc 开源

用少量干声克隆歌手音色,轻松实现歌声转换

lora-svc 是一个基于 Whisper 和 LoRA 技术的歌声转换与克隆开源项目。它解决的核心问题是:如何用···

★ 646 评分 2022-09-08
Dreambooth-Stable-Diffusion 开源

用几张照片,微调Stable Diffusion生成专属主体的新图像

Dreambooth-Stable-Diffusion 是 Dreambooth 论文的 Stable Diffusion 实现,旨在解决文本到图像模···

★ 7734 评分 2022-09-06
stable-dreambooth 开源

用最少代码微调 Stable Diffusion,几张图定制专属模型。

stable-dreambooth 是一个基于 Stable Diffusion 的 Dreambooth 实现,旨在用极简代码让用户微调自···

★ 146 评分 2022-09-04
whisper-vits-japanese 开源

只有音频也能训练日语VITS,Whisper自动生成文本标注

这是一个将Whisper语音识别模型与VITS语音合成模型结合的开源工具,专门用于日语语音合成。它解决的···

★ 161 评分 2022-08-11
dpm-solver 开源

10步内完成扩散模型采样,图像生成提速数倍

DPM-Solver 是 NeurIPS 2022 Oral 论文的官方实现,针对扩散概率模型(DPM)采样速度慢的问题,提出···

★ 1857 评分 2022-08-07
mttl 开源

给大模型挂载可插拔小模块,一个底座跑多个任务

mttl 是一个用于构建模块化语言模型的开源工具,核心思路是将大模型拆解为可组合的模块,并配合参数···

★ 116 评分 2022-07-11
paxml 开源

用JAX训练超大模型,榨干GPU算力,配置灵活不掉队。

Pax是一个基于JAX的机器学习框架,专为训练大规模模型而设计。它解决了在大规模分布式训练中,模型···

★ 561 评分 2022-06-14