Akegarasu/lora-scripts

SD-Trainer. LoRA & Dreambooth training scripts & GUI use kohya-ss's trainer, for diffusion model.

해결하는 문제

Stable Diffusion용 LoRA 및 Dreambooth 모델 훈련 과정을 단순화하여 사용자 친화적인 인터페이스와 사전 구성된 환경을 제공함으로써 수동 스크립트 설정의 복잡성을 피할 수 있도록 합니다.

작동 방식

이 프로젝트는 kohya-ss/sd-scripts를 위한 포괄적인 래퍼 및 GUI로 작동합니다. 데이터셋 준비를 위한 태그 에디터, WD 1.4 Tagger, 훈련 진행 상황 모니터링을 위한 TensorBoard를 하나의 장소에 통합한 WebUI를 제공합니다.

대상 사용자

스크립트를 작성하거나 복잡한 명령줄 인수를 수동으로 관리하지 않고도 자신의 맞춤형 Stable Diffusion 모델을 훈련하고 싶은 아티스트 및 AI 애호가들입니다.

주요 기능

  • 스트리밍된 훈련 스튜디오 경험을 제공하는 통합 WebUI.
  • 데이터셋 레이블링을 더 쉽게 해주는 내장된 WD 1.4 Tagger 및 태그 에디터.
  • 전용 설치 스크립트를 통해 Windows 및 Linux에서의 네이티브 지원.
  • 훈련 메트릭을 추적하기 위한 통합 TensorBoard 지원.

관련

  • Dispatch
    효율적인 Stable Diffusion 파인튜닝을 위한 LoRA 활용

    Hugging Face는 Low-Rank Adaptation(LoRA)을 diffusers 라이브러리에 통합하여, 훨씬 낮은 VRAM 요구량과 3 MB 정도의 작은 모델 가중치로 Stable Diffusion 파인튜닝을 더 빠르게 수행할 수 있게 했습니다.

  • 프로젝트
    hollowstrawberry/kohya-colab

    Stable Diffusion LoRA 모델 훈련 및 데이터셋 준비를 위한 Google Colab 노트북 모음으로, SDXL 지원도 포함됩니다.

  • 프로젝트
    Nerogar/OneTrainer

    데이터셋 준비, 파인튜닝 및 모델 변환 도구를 제공하는 GUI 또는 CLI 기반 확산 모델 종합 교육 스위트입니다.

  • 프로젝트
    ostris/ai-toolkit

    소비자급 하드웨어에서 GUI 또는 CLI를 통해 이미지, 비디오, 오디오 모델을 미세 조정할 수 있는 확산 모델용 올인원 훈련 스위트.

  • 프로젝트
    yolain/ComfyUI-Easy-Use

    ComfyUI‑Easy‑Use 는 ComfyUI의 노드 기반 UI용 진정한 오픈소스 확장 기능으로, 수십 개의 편의성 노드(로더, 프롬프트 도구, LoRA/ControlNet 헬퍼, 노이즈 주입, 이미지 유틸리티, 논리/루프 노드 등)를 제공하여 Stable Diffusion 및 기타 확산 모델 워크플로우 구축을 간소화합니다. 다양한 모델(SD‑1/2, SDXL, SD‑3, Flux, Kolors, Stable Cascade 등)을 지원하며, UI 테마 개선, 문서, 동영상 튜토리얼, 워크플로우 컬렉션도 포함됩니다. 설치는 ComfyUI의 `custom_nodes` 폴더에 `git clone` 한 후 `install.bat`를 실행하는 것으로 간단합니다.