Repolode
ostris

ostris/ai-toolkit

웹 UI나 CLI를 통해 소비자용 GPU에서 이미지, 비디오, 오디오 diffusion 모델을 파인튜닝하는 올인원 학습 도구 모음입니다.

스타
12,378
포크
1,587
열린 이슈
92
마지막 커밋
그저께

어떤 프로젝트인가요?

AI Toolkit은 diffusion 모델을 파인튜닝하기 위한 학습 도구 모음입니다. FLUX, Qwen-Image, SDXL, Wan 2.x, LTX-2 등의 모델 계열을 비롯해 이미지, 지시 기반 편집, 비디오, 오디오 모델을 지원합니다. 소비자용 하드웨어에서도 이러한 모델을 실행하는 것을 목표로 합니다.

학습은 작업을 시작하고 중지하며 모니터링하는 웹 UI에서 실행하거나, run.py에 전달하는 YAML 설정 파일을 사용해 명령줄에서 실행할 수 있습니다. 중단된 경우 마지막 체크포인트부터 학습을 재개합니다. 데이터셋은 이미지와 이에 대응하는 캡션 .txt 파일이 들어 있는 폴더로 구성되며, 이미지를 자동으로 리사이즈하고 버킷으로 분류하므로 수동으로 크롭할 필요가 없습니다.

LoRA 및 LoKr 네트워크를 지원하며, include 및 exclude 필터를 사용해 특정 레이어를 대상으로 지정할 수 있습니다. PyTorch 환경을 구성해 주는 내장 매니저로 설치하거나, Python 3.10+ 및 Nvidia GPU를 사용해 직접 설치할 수 있습니다.

핵심 기능

  • 01하나의 도구 모음에서 이미지, 편집, 비디오, 오디오 diffusion 모델을 학습합니다.
  • 02학습 작업을 시작하고 중지하며 모니터링할 수 있는 웹 UI를 제공합니다.
  • 03YAML 설정 파일을 사용한 명령줄 학습을 지원합니다.
  • 04레이어별 include 및 exclude 필터와 함께 LoRA 및 LoKr 네트워크를 지원합니다.
  • 05중단 후 마지막 체크포인트부터 학습을 재개합니다.
  • 06화면 비율이 서로 다른 이미지를 자동으로 리사이즈하고 버킷으로 분류합니다.

활용 사례

  • 소비자용 GPU 한 대에서 FLUX 또는 SDXL LoRA로 원하는 스타일이나 캐릭터를 학습합니다.
  • 개인 데이터셋으로 Wan 또는 LTX 비디오 모델을 파인튜닝합니다.
  • 원격 서버에서 학습 작업을 실행하고 웹 UI로 모니터링합니다.

빠른 시작

  1. 1

    저장소를 클론합니다

    git clone https://github.com/ostris/ai-toolkit.git
  2. 2

    프로젝트 폴더로 이동합니다

    cd ai-toolkit
  3. 3

    Linux에서 AI Toolkit Manager를 시작합니다(환경을 설정하고 http://localhost:8675 에서 UI를 실행합니다)

    ./run_linux.sh

사용 전 알아둘 점

  • 원클릭 매니저는 제작자가 실험적인 기능으로 표시했습니다.
  • 수동 설치에는 Nvidia GPU와 학습할 모델을 위한 충분한 VRAM이 필요합니다.
  • FLUX.1-dev와 같은 일부 지원 모델은 먼저 Hugging Face에서 액세스를 요청해야 합니다.

토픽

자주 묻는 질문

ai-toolkit은(는) 무엇인가요?
웹 UI나 CLI를 통해 소비자용 GPU에서 이미지, 비디오, 오디오 diffusion 모델을 파인튜닝하는 올인원 학습 도구 모음입니다. AI Toolkit은 diffusion 모델을 파인튜닝하기 위한 학습 도구 모음입니다. FLUX, Qwen-Image, SDXL, Wan 2.x, LTX-2 등의 모델 계열을 비롯해 이미지, 지시 기반 편집, 비디오, 오디오 모델을 지원합니다. 소비자용 하드웨어에서도 이러한 모델을 실행하는 것을 목표로 합니다.
ai-toolkit은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
ai-toolkit은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
ai-toolkit은(는) 아직 유지보수되고 있나요?
활발히 유지보수 중. ai-toolkit의 마지막 커밋은 2026년 10월 19일입니다.
ai-toolkit은(는) 어디에 사용할 수 있나요?
소비자용 GPU 한 대에서 FLUX 또는 SDXL LoRA로 원하는 스타일이나 캐릭터를 학습합니다. 개인 데이터셋으로 Wan 또는 LTX 비디오 모델을 파인튜닝합니다. 원격 서버에서 학습 작업을 실행하고 웹 UI로 모니터링합니다.
ai-toolkit는 어떻게 설치하나요?
저장소를 클론합니다: git clone https://github.com/ostris/ai-toolkit.git 프로젝트 폴더로 이동합니다: cd ai-toolkit Linux에서 AI Toolkit Manager를 시작합니다(환경을 설정하고 http://localhost:8675 에서 UI를 실행합니다): ./run_linux.sh
ai-toolkit은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/ostris/ai-toolkit

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트