ostris/ai-toolkit

무엇이 문제를 해결하나요

AI Toolkit은 확산 모델을 위한 통합적이고 사용하기 쉬운 훈련 세트를 제공하여 사용자가 복잡한 환경 설정을 수동으로 관리하지 않고도 소비자급 하드웨어에서 이미지, 비디오, 오디오 모델을 훈련할 수 있게 합니다.

어떻게 작동하나요

이 툴킷은 FLUX, SDXL, Wan 2.1 등을 포함한 다양한 모델을 지원하며, 훈련 과정과 상호작용하는 두 가지 주요 방법을 제공합니다: YAML 구성 파일을 사용하는 명령줄 인터페이스(CLI)와 작업을 모니터링하고 관리하는 웹 기반 그래픽 사용자 인터페이스(GUI). 또한 PyTorch, Node.js, FFmpeg의 설치를 자동화하는 실험적인 관리자를 포함하여 환경을 시스템으로부터 격리시킵니다.

누구를 위한 도구인가요

이 도구는 LoRA와 LoKr 같은 기술을 사용하여 확산 모델을 미세 조정하려는 창작자와 개발자를 위해 설계되었으며, 특히 하드웨어 자원이 제한적이거나 설정을 간소화하려는 사람들에게 적합합니다.

주요 기능

• 넓은 모델 지원: 이미지, 비디오, 지시/편집, 오디오 확산 모델의 광범위한 목록과 호환됩니다.
• 자동 설정: 실험적인 관리자가 의존성 설치와 환경 생성을 처리합니다.
• 유연한 훈련: LoRA와 LoKr 훈련을 지원하며 특정 모델 레이어를 대상으로 하거나 제외할 수 있습니다.
• 간소화된 데이터 처리: 이미지 크기 조정과 종횡비를 자동으로 처리하며, 이미지와 일치하는 텍스트 캡션 파일만 필요합니다.
• 클라우드 통합: RunPod 및 Modal에서 훈련을 위한 공식 템플릿과 가이드를 제공합니다.