/ Модель

Характеристики

Лицензия
CreativeML OpenRAIL-M
Опубликована
27 окт 2022

sd-tohoku-v1

Генерация изображенийТекст → изображение3 профессии

DreamBooth-дообученная Stable Diffusion для генерации иллюстраций с пятью персонажами проекта Tohoku Zunko по текстовым запросам.

25скачиваний за 30 дней16лайковвеса обновлены27 мар 2023

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Тесты пользователейпока нет

Никто ещё не опубликовал тест этой модели. Соберите бенчмарк из своих задач — результат появится здесь.

  1. Добавьте модель в бенчмарк

    Вместе с другими моделями категории «Изображения» — сравнение идёт внутри одной категории.

  2. Опишите свои задачи

    Напишите запросы, при необходимости приложите файлы и эталонные ответы.

  3. Запустите и опубликуйте

    Модели сгенерируют результат по каждой задаче. Опубликованный тест появится на этой странице.

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

shirayu/sd-tohoku-v1 — модель генерации изображений по текстовому описанию, созданная Shirayu на основе Stable Diffusion с дообучением DreamBooth. Она предназначена для создания иллюстраций с персонажами проекта Tohoku Zunko: Тохоку Итако, Тохоку Дзунко, Тохоку Киритан, Дзунда-мон в человеческом облике и Сикоку Метан. Для задания персонажа используются триггерные слова itako, zunko, kiritan, zundamon и metan; модель также принимает описания сцены и визуального стиля на английском языке.

Обучение проводилось на 69 изображениях размером 512×512, а базой послужила Nilaier/Waifu-Diffusers. Модель доступна в форматах Diffusers и PyTorch; лицензия — CreativeML Open RAIL-M 1.0. Для изображений с персонажами проекта необходимо учитывать правила их использования. В README отмечено, что официальные костюмы могут генерироваться реже, поскольку обучающие изображения включали разнообразные варианты одежды.

Сценарии применения

  • создание иллюстраций с Тохоку Дзунко
  • генерация изображений Тохоку Киритан в заданной сцене
  • создание артов с Дзунда-моном в человеческом облике
  • генерация изображений Тохоку Итако и Сикоку Метан
  • создание фан-арта с персонажами проекта Tohoku Zunko

Ограничения

Официальные костюмы персонажей могут генерироваться реже из-за разнообразия одежды в обучающих изображениях. Использование изображений с персонажами должно соответствовать правилам проекта Tohoku Zunko.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

以下の5人の東北ずん子プロジェクトのキャラクターイラストを用いてDreamBoothで学習したモデルです.

  • itako: 東北イタコ
  • zunko: 東北ずん子
  • kiritan: 東北きりたん
  • zundamon: ずんだもん (人間形態)
  • metan: 四国めたん

学習画像はなるべく衣装にバリエーションをもたせているので,「公式衣装」は出にくいです.

🌐オンラインですぐに試せるGoogle Colab Notebookでも利用できます

🔈 キャラクターを増やして学習したモデルshirayu/sd-tohoku-v2を公開しました (2023-01-04)

ファイル形式

  1. AUTOMATIC1111/stable-diffusion-webuiなどckptファイルを読み込むツールの場合

    sd-tohoku-v1.model.ckpt(約2GB)をダウンロードして読み込んでください

  2. diffusersから利用する場合

    from diffusers import DiffusionPipeline
    pipeline = DiffusionPipeline.from_pretrained("shirayu/sd-tohoku-v1")
    

紹介動画

ライセンス

CreativeML Open RAIL-M license 1.0

また,各種法令・各種ガイドラインにご留意ください. 例えば,生成された画像が東北ずん子プロジェクトのキャラクターを含む場合, 「東北ずん子プロジェクト キャラクター利用の手引き」に基づいて利用してください.

学習設定

  • 元モデル: Nilaier/Waifu-Diffusers (fbd1958)

  • 学習画像

    • 5キャラクター計69枚
      • itako: 東北イタコ 18枚
      • zunko: 東北ずん子 13枚
      • kiritan: 東北きりたん 13枚
      • zundamon: ずんだもん (人間形態) 9枚
      • metan: 四国めたん 16枚
    • アルファチャンネルは削除 + 白背景 + センタリング + 512x512にリサイズ
  • 学習元コード: ShivamShrirao/diffusers (7232c2a)

  • 学習設定

    • Instance ID: itako, kiritan, zunko, metan, zundamon (5種)
    • Instance prompt: <ID> 1girl
    • Tesla T4で約110分
    • その他設定:
    --prior_loss_weight=0.5 \
    --seed=3434554 \
    --resolution=512 \
    --center_crop \
    --train_batch_size=1 \
    --train_text_encoder \
    --mixed_precision="fp16" \
    --use_8bit_adam \
    --gradient_checkpointing \
    --gradient_accumulation_steps=2 \
    --learning_rate=1e-6 \
    --lr_scheduler="constant" \
    --lr_warmup_steps=0 \
    --num_class_images=50 \
    --sample_batch_size=3 \
    --max_train_steps=8000
    

学習に使った画像

生成例

kiritan, 1girl, volleyball, kawaii, in gymnasium, head
Negative prompt: chibi, out of frame, armature drawing, mutated hands and fingers, poor drawing, amateur, bad painting, bad painting of arms, bad anatomy, mutation, extra limbs, ugly, fat
Steps: 40, Sampler: Euler a, CFG scale: 7.5, Seed: 575469807, Size: 704x512
zundamon , maid dress, in cafe, Kyoto Animation
Negative prompt: chibi, out of frame, armature drawing, mutated hands and fingers, poor drawing, amateur, bad painting, bad painting of arms, bad anatomy, mutation, extra limbs, ugly, fat
Steps: 40, Sampler: Euler a, CFG scale: 7.5, Seed: 429473516, Size: 512x704
itako, dating in park, cute winter fashion
Negative prompt: out of frame, amateur drawing, mutated hands and fingers, poor drawing, amateur, bad painting, bad painting of arms, bad anatomy, mutation, extra limbs, ugly, fat
Steps: 60, Sampler: Euler a, CFG scale: 7.5, Seed: 2722676181, Size: 704x512
zunko and metan sit on bench, in school uniform, drink tea, 2girls, in 2020s anime style
Negative prompt: chibi, armature drawing, mutated hands and fingers, poor drawing, amateur, bad painting, bad painting of arms, bad anatomy, mutation, extra limbs, ugly
Steps: 40, Sampler: Euler a, CFG scale: 7.5, Seed: 2262270937, Size: 640x512

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм

Можно запустить

Подбираем GPU…

Ревизия весов
5f54f07

0 моделей