Stable Journey Bot

Discord bot client for AUTOMATIC1111's stable-diffusion-webui.

機能

• Prompt Style のサポート
• プロンプトプリセット
• ネガティブプロンプト
• サンプラーの選択
• モデルの切り替え
• 複数枚生成
• リローリング
• highres fix

使い方

プロジェクトのクローン

レポを落として、ルートに移動します。

git clone https://github.com/p1atdev/stable-journey-bot.git
cd stable-journey-bot

Web UIの準備

必要に合わせて起動フラグに以下を追加します

• --api 必須。API機能を有効にします
• --listen 欲しい場合のみ。0.0.0.0 でホストされるので、ドメインと紐づけている人とかはこれが必要。ただし、これを有効にすると拡張機能を入れることが出来ないので、頻繁に拡張機能を弄る人は非推奨。
• --nowebui 推奨。Gradio の Web UI 機能を無効にします。Bot から生成するだけなら Web UI は不要なので、これを有効にすると起動が早くなったりする気がします。多分。
• --share Botを自分のPC以外で動かしたいときには必須。ただし --nowebui とは共存できません。
• --ngrok Gradio の share を利用しないときの選択肢。詳しい使い方知らないのでわからない。

また、起動時に出る Running on local URL: http://0.0.0.0:7860 や Uvicorn running on http://0.0.0.0:7861 はちゃんと見ておきましょう。特にポート番号。基本的に localhost:7860 ですが、 --nowebui の場合は 7861 になります。

./webui-user.bat の設定例

ローカルで軽く動かすとき

@echo off

set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--xformers --api --nowebui
set ATTN_PRECISION=

call webui.bat

--xformers などはお好みで指定してください。

外部に公開するとき

@echo off

set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--xformers --api --share
set ATTN_PRECISION=

call webui.bat

BOT の設定

config.example.yaml をコピー&リネームして config.yaml を作成します。

例に従って必要な項目を入力します。

# Discord
DISCORD_TOKEN: hogehoge
GUILD_ID: "1234567890"

# BOT
globalCommands: false
allows:
  - switch
  - refresh
  - imagine
  - info
  - status
  # - neko

defaultParameters:
  style: ""
  sampler: "DPM++ 2M Karras"
  width: 512
  height: 768
  highresFix: false 
  clipSkip: 2 

additionalParameters:
  promptPrefix: "masterpiece, exceptional"
  promptSuffix: "ultra detailed"
  negativePromptPrefix: "NSFW, badquality, bad anatomy"
  negativePromptSuffix: ""

serverType: "AUTOMATIC1111"

# API server address
host: http://localhost:7860

• DISCORD_TOKEN: BOTのトークン

• GUILD_ID: BOTを動かすサーバー

• globalCommands: グローバルでスラッシュコマンドを有効にするかどうか

• allows: 実行を許可するコマンド名

• defaultParameters: デフォルトのパラメーター。 /imagine で指定されない場合は、この値が使われる。

  • style: デフォルトで使用する Prompt Style の名前
  • widht: 幅
  • height: 高さ
  • highresFix: highres fix を有効にする
  • ckipSkip: CLIP skip の値を変更する

• additionalParameters:

  • promptPrefix: 常に prompt の先頭に追加する内容
  • promptSuffix: 常に prompt の後ろに追加する内容
  • negativePromptPrefix: 常に negative prompt の先頭に追加する内容
  • negativePromptSuffix: 常に negative prompt の後ろに追加する内容

• serverType: 現状は常に AUTOMATIC1111。いつかほかのに対応するかも？しないかも？

• host: Web UI のアドレス

実行

deno task start

で実行します。Ctrl + C で停止です。

実行 (docker compose)

docker compose up

コマンド一覧

/switch

モデルを切り替えます。

• name: モデル名 (必須)

/refresh

モデルなどをリフレッシュします。

Stable-diffusion ディレクトリにモデルを追加したり削除した場合は、これを実行しないとモデルを切り替えることが出来ません。

/imagine

生成します。

• prompt: ポジティブプロンプト (必須)
• negative: ネガティブプロンプト
• prompt-style: プロンプトスタイル。入力されたプロンプトの先頭に挿入されます。
• width: 幅。512, 768, ~ 2048 まででキリがいいものを入れてます。
• height: 高さ。width 同様
• seed: シード値
• sampler: サンプラー
• steps: ステップ数
• scale: CFG scale
• highres-fix: highres fix を有効にします
• clip-skip: CLIP skip を変更します
• count: 生成枚数。最大は 4。

TODO

• [x] /imagine のデフォルト値の設定
• [x] アスペクト比や生成枚数をカスタマイズできるように
• [ ] 生成したものの削除
• [ ] エラーハンドリング
• [ ] リローリング、バリエーション (midjourney風に)
• [ ] img2img
• [ ] Lsmith?