Guest User

Untitled

a guest
Jul 10th, 2026
76
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 4.43 KB | None | 0 0
  1. Твоя задача — реализовать удобную и стабильную реализацию метода учитель–ученик для обучения LoRA и full моделей с использованием DINOv3 моделей. Основной фокус — пайплайн обучения для SDXL (как основная тестовая ветка данного метода).
  2.  
  3. ---
  4.  
  5. ## КОНТЕКСТ
  6.  
  7. Тебе предоставлен весь репозиторий:
  8. https://github.com/kohya-ss/sd-scripts
  9.  
  10. ---
  11.  
  12. ## ОСНОВНОЙ МЕТОД
  13.  
  14. Метод основан на подходе self-distillation (teacher–student) без разметки, описанном здесь:
  15.  
  16. https://jehillparikh.medium.com/improved-image-encoding-using-transformers-self-distillation-with-no-labels-dino-dinov2-79ac5b6cac06
  17.  
  18. ---
  19.  
  20. ## ТРЕБОВАНИЯ К РЕАЛИЗАЦИИ
  21.  
  22. ### 1. Общие требования
  23.  
  24. * Реализовать training pipeline teacher–student для SDXL
  25. * Поддержка обучения LoRA и full fine-tuning моделей
  26. * Использование DINOv3 / DINO-style моделей в роли teacher
  27. * Student обучается через distillation loss
  28.  
  29. ---
  30.  
  31. ### 2. Ограничения по ресурсам
  32.  
  33. Обучение должно быть рассчитано на consumer GPU:
  34.  
  35. * GPU: 12GB VRAM
  36. * RAM: 32GB
  37.  
  38. Обязательно учитывать:
  39.  
  40. * минимизацию потребления памяти
  41. * возможность использования caching для ускорения обучения и снижения нагрузки
  42.  
  43. ---
  44.  
  45. ### 3. Loss functions
  46.  
  47. Должна сохраняться возможность выбора и расширения loss функций:
  48.  
  49. * DINO-based loss (основной вариант)
  50. * стандартный loss (из текущего репозитория)
  51. * возможность добавления кастомных loss функций без изменения основной архитектуры
  52.  
  53. ---
  54.  
  55. ### 4. Интеграция в репозиторий
  56.  
  57. * Не ломать существующий функционал https://github.com/kohya-ss/sd-scripts
  58. * Сохранить весь текущий training pipeline без изменений поведения по умолчанию
  59. * Добавить метод как отдельный модуль / режим обучения
  60. * Интеграция должна быть максимально модульной
  61.  
  62. ---
  63.  
  64. ### 5. Модульность
  65.  
  66. Реализация должна быть разделена на независимые компоненты, чтобы:
  67.  
  68. * можно было исправлять отдельные части без изменения всей системы
  69. * можно было отключить DINO pipeline и использовать стандартный training
  70.  
  71. ---
  72.  
  73. ### 6. Caching
  74.  
  75. Необходимо предусмотреть систему кеширования:
  76.  
  77. * возможность хранения промежуточных результатов teacher модели
  78. * использование кеша для ускорения обучения
  79. * совместимость с датасетом и конфигурацией обучения
  80.  
  81. ---
  82.  
  83. ## ДОПОЛНИТЕЛЬНЫЕ ИСТОЧНИКИ
  84.  
  85. Для понимания структуры и стиля проекта использовать:
  86.  
  87. * CLAUDE.md
  88. * EXAMPLES.md
  89.  
  90. Ресурсы Meta (DINOv3):
  91.  
  92. * https://ai.meta.com/research/publications/dinov3/
  93. * https://github.com/facebookresearch/dinov3
  94.  
  95. ---
  96.  
  97. ## КРИТЕРИИ УСПЕШНОЙ РЕАЛИЗАЦИИ
  98.  
  99. * Работает обучение SDXL LoRA (и LyCORIS https://github.com/KohakuBlueleaf/LyCORIS - в основном будет использоваться он, репозиторий сейчас работает с LyCORIS) с DINO distillation
  100. * Работает full fine-tuning SDXL с тем же подходом
  101. * Поддерживается caching
  102. * Поддерживается pluggable loss system
  103. * Не нарушен существующий функционал sd-scripts
  104. * Реализация модульная и легко расширяемая
  105. * Работает на 12GB GPU без критических OOM при LoRA обучении
  106.  
Advertisement
Add Comment
Please, Sign In to add comment