mp3pintyo/ai-toolkit-template

镜像概述

本镜像为Mp3Pintyo开发的AI工具包，专注于Flux.1-Dev模型的LoRA训练任务，提供便捷的环境配置与训练支持，帮助用户快速开展模型微调工作。

核心功能

• 支持Flux.1-Dev模型的LoRA训练流程
• 简化训练环境搭建步骤
• 提供基础训练参数调整选项

使用场景

• AI模型开发者对Flux.1-Dev模型进行特定风格微调
• 研究人员探索LoRA训练在生成式AI中的应用场景
• 个人用户尝试自定义Flux.1-Dev模型的输出效果

配置说明

• 通过环境变量可调整训练批次大小、学习率等核心参数
• 需要挂载训练数据目录与输出结果目录到容器内对应路径

部署示例

Docker Run命令

bash
docker run -v /本地训练数据路径:/data -v /本地输出路径:/output docker.xuanyuan.run/mp3pintyo/ai-toolkit-template

Docker Compose示例

yaml
version: '3'
services:
  ai-toolkit:
    image: mp3pintyo/ai-toolkit-template
    volumes:
      - ./local_data:/data
      - ./local_output:/output
    environment:
      - BATCH_SIZE=8
      - LEARNING_RATE=1e-4