AI Toolkit (Web UI版) をGoogle Colabで動かすためのノートブックとなります

aitoolkit_colab.sh

#@markdown # AI Toolkit (Web UI版) をGoogle Colabで動かすためのノートブック
#@markdown ## Hugging FaceのAccess Tokens設定
#@markdown FLUX.1-devでトレーニングを行うには必須となります。Access TokensはHugging Faceのサイトにログインした状態で以下のリンクから取得できます。

#@markdown https://huggingface.co/settings/tokens

#@markdown **なおHugging FaceのAccess Tokensは人と共有しないようにお気をつけください**

HUGGINGFACE_TOKEN= "" #@param {type:"string"}

#@markdown ---

# Hugging Faceログイン
!huggingface-cli login --token {HUGGINGFACE_TOKEN}

# AI Toolkitの設定・起動
!git clone https://github.com/ostris/ai-toolkit.git
%cd ai-toolkit
!git submodule update --init --recursive

# sqlite3があるとエラーになるため事前に削除
!sed -i '/^sqlite3.*$/d' requirements.txt

!pip3 install -r requirements.txt
!python flux_train_ui.py