Nerogar/OneTrainer

OneTrainer is a one-stop solution for all your Diffusion training needs.

What it solves

OneTrainer는 확산 모델의 학습 및 파인튜닝을 위한 포괄적인 올인원 솔루션을 제공합니다. 데이터셋 준비, 학습 실행 관리 및 모델 포맷 변환 과정을 단순화하여 여러 개별 도구를 전환할 필요를 없애줍니다.

How it works

이 소프트웨어는 그래픽 사용자 인터페이스(GUI)와 명령줄 인터페이스(CLI)를 모두 제공하여 학습 파이프라인을 관리합니다. FLUX.1, Stable Diffusion 1.5~3.5, Hunyuan Video 등 다양한 확산 모델을 지원하며, 전체 파인튜닝, LoRA, 임베딩 등 여러 학습 방식을 지원합니다. 자동 캡션(BLIP, BLIP2, WD-1.4) 및 마스크 생성(ClipSeg, Rembg) 도구가 내장되어 있어 데이터셋 준비가 원활합니다.

Who it’s for

이 도구는 AI 아티스트와 개발자를 위해 설계되었으며, 코딩 지식이 많지 않아도 이미지·비디오 생성 모델을 높은 수준으로 제어하면서 파인튜닝할 수 있게 해줍니다.

Highlights

  • Broad Model Support: SDXL, Stable Cascade, PixArt 등 광범위한 모델과 호환.
  • Flexible Training: 전체 파인튜닝, LoRA, 임베딩을 지원하며 마스크 학습 및 다중 해상도 학습 옵션 제공.
  • Dataset Tooling: 자동 캡션 및 마스크 생성 도구 통합.
  • Integrated Sampling: UI 내에서 직접 모델을 샘플링하여 학습 진행 상황을 실시간 모니터링 가능.
  • Training Optimizations: 종횡비 버킷팅, EMA(지수 이동 평균) 지원, 노이즈 스케줄러 리스케일링 기능 포함.