ChatR Labs Uncensored Omni LoRA v0.1

Экспериментальный rank-128 LoRA-адаптер, дообученный от Qwen/Qwen-Image-2.1. Работает и для генерации по тексту, и для редактирования изображений по инструкции на естественном языке. В релизе только веса LoRA — базовая модель, text encoder, processor и VAE загружаются отдельно.

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

• 🧩 Один адаптер для text-to-image и image editing; в Playground проверен с 1 и 3 референсами.
• 📀 BF16-адаптер на 671 MB в Diffusers/PEFT-ключах; без merged base weights, GGUF/FP8.
• ⚙ Рекомендуемые настройки: LoRA strength 1.0, 35 steps, true CFG 5.0, negative prompt — один пробел, ~1 MP, KV cache включён.
• ⚡ Rank/alpha 128/128, 224 модуля в 32 transformer blocks; обучение на 1024×1024 с aspect-ratio bucketing.
• 🛑 Это обычный LoRA, не few-step distillation; несовместим с ранними Qwen-Image, Qwen-Image-2512 и Qwen-Image-Edit-2511.

💡 Пример

import torch
from diffusers import QwenImage21Pipeline

pipe = QwenImage21Pipeline.from_pretrained('Qwen/Qwen-Image-2.1', torch_dtype=torch.bfloat16)
pipe.load_lora_weights('chatrlabs/Qwen-Image-2.1-Uncensored-Omni-LoRA-v0.1',
                       weight_name='chatrlabs-omni-lora-v0.1-bf16.safetensors', adapter_name='omni')
pipe.set_adapters('omni', adapter_weights=1.0)
image = pipe(prompt='A blue ceramic mug', negative_prompt=' ', num_inference_steps=35).images[0]

Подробнее

Качественных бенчмарков в v0.1 нет; замеры деплоя: 33,2 с (t2i 1344×768), 38,4 с (edit, 1 ref), 50,5 с (edit, 3 refs) при 25 steps и CFG 4.0; при 35 steps ожидайте ~1,4×.
Playground использует две RTX 4090 24 GB с разделением BF16-пайплайна; при нехватке VRAM нужен CPU offload или multi-GPU.
Чекпойнт: step 32 000 (около 1,75 из 8 эпох), 585 643 training items, rank/alpha 128/128, 224 модуля в 32 transformer blocks; датасет не опубликован.
«Uncensored»: адаптер и пример кода не добавляют prompt filter или output safety checker, но это не гарантия отрисовки любого запроса.
Лицензия Qwen Research License: только некоммерческие исследования и оценка без отдельной лицензии.