Welmia Face 2.0

Welmia Face 2.0 Base — небольшая безусловная GAN в стиле StyleGAN2 от Welmia на 7 625 453 параметра генератора. Генерирует лица 128×128 RGB из случайного шума, весит около 30,5 МБ в float32 и запускается на обычном CPU без GPU.

💻 Ключевые особенности

• 💻 StyleGAN2-подобный генератор: mapping network, модулированные свёртки с демодуляцией, шум по слоям, skip-RGB, без BatchNorm.
• 👁 Выход 128×128 RGB; латент z и стиль w — по 256.
• 🔥 На Intel Core i5-10400 медиана ~22 img/s и ~46 мс на изображение; замер только forward pass.
• 🎨 Параметр psi управляет компромиссом между чистотой и разнообразием: 0.5 — чище, 1.0 — разнообразнее, но больше артефактов.
• ✅ Веса в safetensors; каждый запуск печатает seed для воспроизведения.

💻 Пример

import json, sys, torch
from huggingface_hub import snapshot_download
from safetensors.torch import load_file
path = snapshot_download("Welmia/welmia-face-2.0-7.6m-base")
sys.path.insert(0, path)
import welmia_gan as W
cfg = json.load(open(f"{path}/welmia_cfg.json")); g = W.make_G(cfg)
g.load_state_dict(load_file(f"{path}/model.safetensors")); g.eval()

Подробнее

Обучение с нуля на 2 GPU, batch 64, Adam с lr 0.002; в репозитории только EMA-генератор.
Замер на i5-10400/16 ГБ без GPU: медиана ~22 img/s и ~46 мс на изображение, без сохранения PNG.
Ограничения: 128×128, артефакты фона/волос/аксессуаров, смещение датасета.
Лицензия CC BY-NC-SA 4.0: только некоммерческое использование с атрибуцией Welmia.