Stable Diffusion 등 이미지 생성 모델의 학습·생성·유틸리티 스크립트 모음으로, LoRA와 파인튜닝을 지원합니다.
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
저장소를 클론합니다
git clone https://github.com/ostris/ai-toolkit.git - 2
프로젝트 폴더로 이동합니다
cd ai-toolkit - 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 통계는 매일 갱신됩니다.
관련 프로젝트
이미지·비디오·오디오·3D 콘텐츠 생성을 위해 로컬에서 diffusion 모델을 실행하는 노드 그래프 GUI, API 및 백엔드입니다.
저VRAM 추론과 학습을 지원하는 이미지·비디오·오디오 생성용 오픈소스 Diffusion 모델 엔진입니다.
대형 언어 모델을 레이어별로 실행해 양자화 없이도 70B급 모델을 단일 4GB GPU에서 추론할 수 있는 Python 라이브러리입니다.
이미지·비디오·오디오 생성을 위한 사전 학습 diffusion 모델, 노이즈 스케줄러 및 구성 요소를 제공하는 PyTorch 라이브러리입니다.
YAML 파일로 설정하는 무료 오픈소스 프레임워크로, 대규모 언어 모델의 미세 조정과 후속 학습을 지원합니다.