从24GB显卡到云端用AI Toolkit十分钟跑通FLUX LoRA训练【免费下载链接】ai-toolkitThe ultimate training toolkit for finetuning diffusion models项目地址: https://gitcode.com/GitHub_Trending/ai/ai-toolkit想给自己的角色训一个LoRA却被一屏幕的YAML参数劝退其实 AI Toolkit LoRA 训练没想象中可怕克隆仓库、打开网页界面、传图、填触发词它就开始跑了。FLUX、SDXL、Qwen-Image、Wan视频模型30多个主流扩散模型都在支持列表里。你只需要准备一文件夹图片和一张显卡。先别急着碰参数看看它到底能做什么。能训哪些模型一张清单看懂图像模型FLUX.1/FLUX.2、SDXL、SD 1.5、Qwen-Image、HiDream、OmniGen2 等 30 多个挑一个就能训 LoRA指令编辑模型FLUX.1-Kontext、Qwen-Image-Edit能练听指令改图的能力视频模型Wan 2.1/2.2、LTX-2、MiniMax图生视频的 LoRA 也支持音频模型Ace-Step音乐生成同样可以微调。✨这些模型都支持 8bit 量化和多分辨率分桶不同尺寸的图片丢进同一个文件夹就行系统自动帮你缩放归类。三步跑起来从克隆到打开界面第一步装环境和依赖。你只需复制下面这段命令它会把 Python 环境、PyTorch、全部依赖一次装好git clone https://gitcode.com/GitHub_Trending/ai/ai-toolkit cd ai-toolkit python3 -m venv venv source venv/bin/activate pip3 install torch torchvision torchaudio pip3 install -r requirements.txt第二步启动 Web 界面需要 Node.js 20 以上。界面只负责启动和盯梢任务任务跑起来后关掉它也不影响训练cd ui npm run build_and_start第三步浏览器打开localhost:8675选模型、上传你的图片、填上触发词点开始第一个 LoRA 就上路了。️不同显卡选哪份配置别猜了看这里配置文件都在示例目录里每行都带注释复制一份改改就能用。三种常见场景你的情况推荐配置一句话说明24GB 显卡追求最好画质FLUX.1-dev 训练配置8bit 量化梯度检查点已配好24GB 稳稳跑想商用、推理快FLUX.1-schnell 配置Apache 2.0 许可商用也安心想试试视频 LoRAWan 2.1 1B 配置参数量小、显存压力低适合首次尝试改完存到 config 目录用python run.py 你的配置.yml启动输出文件夹里会自动出现检查点和每隔 250 步的采样图进度一目了然。让训练效果再进一步几个关键开关显存爆掉示例配置里quantize: true默认就是开的这是 8bit 混合精度再配合gradient_checkpointing24GB 显卡跑 FLUX 毫无压力。实在 OOM就取消注释low_vram慢一点但更稳。触发词怎么写填到trigger_word里建议用 p3r5on 这种不易和现有词撞车的写法。系统会自动把它追加到每张图的描述里不用你手动改文本。配文太麻烦每张图旁边放一个同名.txt写一句描述就够再懒一点直接用界面里的 AI 自动配文功能整批图它替你搞定。你的第一步从这里开始别再被参数吓住了把 24GB 那份示例配置复制出来改掉模型路径和图片文件夹就能跑通你的第一个 AI Toolkit LoRA 训练任务。先跑通再谈完美剩下的细节都可以慢慢调。随时欢迎上手试试 【免费下载链接】ai-toolkitThe ultimate training toolkit for finetuning diffusion models项目地址: https://gitcode.com/GitHub_Trending/ai/ai-toolkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
