Kontext-Watermark-Remover

Kontext-Watermark-Remover — LoRA-адаптер для black-forest-labs/FLUX.1-Kontext-dev. Он убирает водяные знаки и текстовые логотипы с изображений, сохраняя фон, текстуры, освещение и общую реалистичность. Обучен на 150 парах изображений: 75 исходных и 75 обработанных.

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

• 👁 Точечно удаляет watermark-текст и логотипы, стараясь не оставлять видимых следов.
• ✨ Сохраняет контекст сцены: фон, текстуры и освещение.
• 💻 Загружается как LoRA-адаптер в diffusers поверх FluxKontextPipeline.
• ➡ Использует подробный триггерный промпт с [photo content] и указанием удалить watermark.
• 🎨 В обучении: 25 эпох, network dimension 64, alpha 32, оптимизатор AdamW8bit.

🧃 Пример

from diffusers import FluxKontextPipeline
pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
pipe.load_lora_weights("prithivMLmods/Kontext-Watermark-Remover", weight_name="Kontext-Watermark-Remover.safetensors")

Подробнее

Базовая модель: FLUX.1-Kontext-dev, в настройках указан вариант fp8.
Обучение: 150 пар изображений, 25 эпох, 2900 шагов + 400 warm-up; UNet LR 5e-5, text encoder LR 1e-5.
В карточке нет бенчмарков и требований к VRAM; подписи для обучения — на английском.