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をバンドルした自己完結型環境を探している人。
  • 学生:拡散トレーニングの実践的なプレイグラウンドを求め、ブラウザで結果を確認したい人。

はじめに(簡単)

  1. リポジトリをクローンします。
  2. プラットフォーム固有のrun_*.sh/.batスクリプトを実行します。マネージャーが仮想環境を作成し、CUDA対応の適切なPyTorchをインストールし、UIを起動します。
  3. config/examples/から設定を選択し(例:train_lora_flux_24gb.yaml)、パス/キャプションを編集して、次を実行します:
    python run.py config/your_config.yml
    
  4. ウェブUIまたはターミナルで進捗を監視します。

制限事項 / 注意点

  • AI Toolkit Managerは実験的とマークされています。手動インストール手順にフォールバックすることもできます。
  • JPEG/PNG画像のみサポート。WebPは現在壊れています。
  • UIはオプションです。トレーニングは完全にスクリプト化できます。
  • 一部のモデル(例:FLUX.1-dev)にはHugging Faceアクセストークンが必要です。

ライセンスとコミュニティ

プロジェクトは無料でオープンソースです(READMEにライセンスは明記されていません)。作者はサポートのためにDiscordへの参加を推奨し、継続的な開発を支援したい人のためのスポンサーシップオプションをリストしています。


要約: Ostris AI Toolkitは、拡散モデルのための本格的で機能豊富なトレーニングスイートであり、便利なウェブUIと完全なCLI制御の両方を提供し、広範なモデルサポートとクラウドデプロイガイドを備えています。


上記の情報はすべてリポジトリのREADMEから直接取得したものであり、外部の推測は追加されていません。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト