ostris/ai-toolkit
The ultimate training toolkit for finetuning diffusion models
Ostris AI Toolkitとは?
Ostris AI Toolkitは、拡散モデル(画像、動画、音声、LLM)をトレーニングおよび実行できるオープンソースのスイートで、単一のマシンまたはクラウドGPU上で動作します。次の機能を提供します:
- GUIウェブUI(
http://localhost:8675で提供)で、トレーニングジョブの開始、停止、監視が可能。 - 同じタスクのためのCLI。ヘッドレスサーバーで便利。
- **実験的な「AI Toolkit Manager」**が、ハードウェアを自動検出し、適切なPyTorchビルドをインストールし、仮想環境を作成し、Node.jsとFFmpegをバンドルするため、システム全体に何もインストールされません。
- Hugging Faceからの多数の事前トレーニング済み拡散モデル(FLUX、SDXL、Qwen-Imageなど)をサポートし、画像生成、命令ベースの編集、動画生成、音声生成、さらにはマルチモーダルLLMをカバー。
- LoRAまたはLoKrアダプターをトレーニングし、特定のレイヤーをターゲットにし、ハイパーパラメーターをカスタマイズできる設定ファイル(YAML)。
- RunPodやModalなどのクラウドプラットフォーム用の統合スクリプト、および作者自身のOstris Cloudサービスへのリンク。
このツールキットは、消費者向けGPUでの「ワンクリック」体験を求めつつ、パワーユーザーが必要とするすべての調整機能を提供することを目的とした、趣味人や研究者を対象としています。
主な機能
| 機能 | 詳細 |
|---|---|
| モデルカタログ | 40以上の拡散モデル(画像、動画、音声)と1つのLLMがリストされ、それぞれHugging Faceリポジトリにリンクされています。 |
| トレーニングUI | Node.jsで構築されたウェブUI。リモートアクセス用のトークンベースの認証。 |
| CLI / マネージャー | `python -m manager install |
| LoRA / LoKrサポート | 設定駆動のアダプター。only_if_contains / ignore_if_containsによるレイヤーターゲティング。 |
| クロスプラットフォーム | Linux、macOS(Apple Silicon、実験的)、Windows用のスクリプト。 |
| クラウド対応 | 既製のRunPodポッドテンプレート、Modalデプロイガイド、オプションのOstris Cloud GPUレンタル。 |
| データセット処理 | シンプルなフォルダーに*.jpg/pngと対応する*.txtキャプション。自動バケット化とリサイズ。 |
| セキュリティ | UIトークン(AI_TOOLKIT_AUTH)で公開サーバーを保護。 |
誰が使うのか?
- アーティスト&クリエイター:自分たちのハードウェアで拡散モデルを微調整したいが、ボイラープレートコードを書きたくない人。
- 研究者:最新のFLUXやSDXLチェックポイントでLoRA/LoKrをすぐに実験したい人。
- 開発者:PyTorch、Node、FFmpeg、UIをバンドルした自己完結型環境を探している人。
- 学生:拡散トレーニングの実践的なプレイグラウンドを求め、ブラウザで結果を確認したい人。
はじめに(簡単)
- リポジトリをクローンします。
- プラットフォーム固有の
run_*.sh/.batスクリプトを実行します。マネージャーが仮想環境を作成し、CUDA対応の適切なPyTorchをインストールし、UIを起動します。 config/examples/から設定を選択し(例:train_lora_flux_24gb.yaml)、パス/キャプションを編集して、次を実行します:python run.py config/your_config.yml- ウェブUIまたはターミナルで進捗を監視します。
制限事項 / 注意点
- AI Toolkit Managerは実験的とマークされています。手動インストール手順にフォールバックすることもできます。
- JPEG/PNG画像のみサポート。WebPは現在壊れています。
- UIはオプションです。トレーニングは完全にスクリプト化できます。
- 一部のモデル(例:FLUX.1-dev)にはHugging Faceアクセストークンが必要です。
ライセンスとコミュニティ
プロジェクトは無料でオープンソースです(READMEにライセンスは明記されていません)。作者はサポートのためにDiscordへの参加を推奨し、継続的な開発を支援したい人のためのスポンサーシップオプションをリストしています。
要約: Ostris AI Toolkitは、拡散モデルのための本格的で機能豊富なトレーニングスイートであり、便利なウェブUIと完全なCLI制御の両方を提供し、広範なモデルサポートとクラウドデプロイガイドを備えています。
上記の情報はすべてリポジトリのREADMEから直接取得したものであり、外部の推測は追加されていません。
関連
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト