Пользователь Reddit описал обучение LoRA для Krea 2 на RTX 3060 с 12 ГБ VRAM

Пользователь рассказал о своём опыте обучения LoRA для модели Krea 2 на видеокарте RTX 3060 с 12 ГБ VRAM и 64 ГБ оперативной памяти. По его словам, на эксперименты ушло около трёх дней. Как пишет r/StableDiffusion (Reddit), распространённые рекомендации по обучению LoKr на видеокартах с 16 ГБ памяти здесь не сработали: AI Toolkit завершался с ошибкой нехватки памяти ещё до первого шага обучения.

В итоге обучение всё же удалось запустить, хотя и с ограничениями. Вместо LoKr использовали LoRA, часть данных выгружали в оперативную память, а разрешение пришлось снизить до 768. При более высоком значении появлялась ошибка OOM. Генерацию примеров во время обучения тоже отключили, она дополнительно расходовала VRAM и приводила к сбоям. По описанию автора, один цикл обучения LoRA при такой конфигурации занимает около 8 часов.

Для датасета взяли 25 фотографий с разными позами, углами съёмки и фонами. К каждому изображению добавляли текстовое описание в отдельном .txt‑файле. Автор подчёркивает, что для лучшего результата важно фиксировать не только ракурс, но и положение головы, например профиль или наклон. Он также пишет, что незадолго до публикации узнал о поддержке Krea 2 в Musubi Tuner. По его предположению, там обучение может работать эффективнее благодаря другой системе offloading.