2022-08-23

HなStable Diffusion

前提として、Stable Diffusionでエロ画像を出そうとしてもsafety checkerという機能が入っており、センシティブ画像を出そうとすると黒塗りになる。

(Stable DiffusionのSaaSであるDream Studioはぼかしだが、多分別技術)

https://github.com/huggingface/diffusers/releases/tag/v0.2.3

そこでGoogle Colabでちゃちゃっと環境を作り、なおかつNSFW回避する。

1. 下記のリンクノートを開く

https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/stable_diffusion.ipynb

2. 下記の箇所を書き換える

vvvvvvvvvvvvvvvvvv

from diffusers import StableDiffusionPipeline

^^^^^^^^^^^^

この一行を書き換えて自前のStable Diffusion Pipelineをクラス定義する。

https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

をこぴってきてL157行目~159行目を消して貼り付ける。

https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py#L157-L159

3. Google Colabの上から順番に実行する

これだけだ。だが、自分性癖に刺さるStable Diffusionの作成は難しい。つーかマジ安定しない。waifuを探したければ、多分Stable Diffusionは合わない。hentai御用達ワードもなかなかヒットしなかったのでムズイ。

一応、redditを参考にワイが発掘したpromptを置いておく。

"full page antique lithograph of naked girl, sexual position, White background, art print, clean brush stroke, realistic highly detailed, post-processing highly detailed, rendered by octane engine, esty"

naked girlの間に年齢を指定するとガチあかんやつ。人の顔を安定して出すのに"lithograph of" はかなり使える。

"nude painting, big breasts, hot petite, long braided hair, hazel eyes, full round face, short smile, cinematic lightning, medium shot, mid-shot, cinematic wallpaper -C 13"

おっぱいの大きさに定評がある白人女性がたくさん出てくる。顔の部位を丁寧に指定することで安定性が増すらしい。

  • from diffusers import StableDiffusionPipeline をgithubのファイルの内容そのままに置き換えても from ...models import AutoencoderKL, UNet2DConditionModel 以下4行でエラー出るんすけど

    • 動いた参考例 10-12行目の以下、「...」を「diffusers.」に from ...models import AutoencoderKL, UNet2DConditionModel from ...pipeline_utils import DiffusionPipeline from ...schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMSch...

      • diff取ったら確かに色々直してたわ。 すまんな、(ノ∀`)タハー

記事への反応(ブックマークコメント)

ログイン ユーザー登録
ようこそ ゲスト さん