Tavris1/AI-Toolkit-Easy-Install

解決する問題

拡散モデルのトレーニング環境のセットアップは、Python、PyTorch、CUDA、FlashAttention、FFmpeg などのバージョンを一致させる必要があり、非常に面倒です。このプロジェクトは、これらの依存関係を単一のダウンロード可能な ZIP にパッケージ化し、その中に 1 つの .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 用）で破損を回避。
• ランチャーの利便性 – サーバーが起動したらブラウザを開き、更新を通知します。
• 安全な使用ガイダンス – 管理者としての実行、システムフォルダー、パス内のスペースを避けるよう警告します。