必要なライブラリ

• diffusers
• discord
• torch類(pytorchサイトからpip)
• transformers
• accelerate
• omegaconf

各種変換

ckpt -> diffusers

python .\convert_original_stable_diffusion_to_diffusers.py --checkpoint_path "./path/to/ckpt" --dump_path "./path/to/models/<output>"

safetensors -> ckpt

ツールを使わせてもらう

VAEの入れ替え

スクリプトを使わせてもらう

python merge_vae.py <挿入先ckpt> <入れ替えるVAE> <出力先>

nsfw無効化

stable-diffusion-v2はプログラム内で無効化できないので diffusersを改変する

$ python
>>> import diffusers
>>> print(diffusers.__file__)

でフォルダの場所を見つけて、diffusers/pipelines/stable_diffusion/safety_check.py内の記述をコメントアウト

for idx, has_nsfw_concept in enumerate(has_nsfw_concepts):
    if has_nsfw_concept:
        if torch.is_tensor(images) or torch.is_tensor(images[0]):
            images[idx] = torch.zeros_like(images[idx])  # black image
        else:
            images[idx] = np.zeros(images[idx].shape)  # black image

if any(has_nsfw_concepts):
    logger.warning(
        "Potential NSFW content was detected in one or more images. A black image will be returned instead."
        " Try again with a different prompt and/or seed."
    )