Tavris1/AI-Toolkit-Easy-Install

해결하는 문제

확산 모델 훈련 환경을 설정하는 것은 악명 높을 정도로 까다롭습니다. Python, PyTorch, CUDA, FlashAttention, FFmpeg 등의 일치하는 버전이 필요합니다. 이 프로젝트는 이러한 종속성을 단일 다운로드 가능한 ZIP에 패키징하고, 그 안에 하나의 .bat 파일을 포함합니다. 실행하면 필요한 모든 것을 설치하거나 업데이트하고 시스템 Python을 건드리거나 가상 환경을 만들지 않고 AI-Toolkit GUI/로컬 서버를 시작합니다.

작동 방식

릴리스 ZIP을 다운로드하고 배치 파일을 새 폴더에 추출한 다음 더블 클릭합니다. 설치 프로그램은 Git, Node.js, FFmpeg, 내장 Python 3.12.10, CUDA 13.0이 포함된 PyTorch 2.13.0, Triton, FlashAttention, torchcodec, huggingface_hub 확장, 마지막으로 최신 AI-Toolkit 자체를 확인하고 설치합니다. 준비가 되면 실행기가 서버를 시작하고, 서버가 활성화되는 즉시 기본 브라우저를 열고 업데이트를 확인합니다. 별도의 Update-AI-Toolkit 스크립트가 이후 업그레이드를 처리합니다.

대상 사용자

NVIDIA GPU가 있는 Windows 사용자로, 패키지 관리자, CUDA 버전 또는 가상 환경을 수동으로 다루지 않고 이미지 및 비디오 확산 모델을 훈련하거나 미세 조정하려는 사람. 휴대용 자체 포함 설정을 선호하는 초보자와 숙련된 사용자 모두를 대상으로 합니다.

하이라이트

• 원클릭 휴대용 설치 프로그램 – 시스템 Python 없음, VENV 없음, 배치 파일 실행만 하면 됩니다.
• 전체 스택 자동 프로비저닝 – Git, Node.js, FFmpeg, Python, PyTorch, Triton, FlashAttention, torchcodec 및 AI-Toolkit 자체.
• 고정된 호환 버전 – FlashAttention 2.8.3, Triton 3.7.1.post27, torchcodec 0.15.0(Torch 2.13.0용)으로 손상 방지.
• 실행기 편의성 – 서버가 켜지면 브라우저를 열고 업데이트를 알려줍니다.
• 안전한 사용 지침 – 관리자로 실행, 시스템 폴더, 경로의 공백에 대해 경고합니다.