kohya-ss / kohya-ss/sd-scripts

[SDXL] Train SDXL refiner for subject generation

Open
#620 9 comments 27 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
7.2k
Forks
1.2k
Avg merge
11m
Merged PRs (30d)
2

Description

I've been using the scripts here to fine tune the base SDXL model for subject driven generation to good effect. However, I've found that adding the refiner step usually means that the refiner doesn't understand the subject, which often makes using the refiner worse with subject generation.

Currently the scripts do not support fine tuning the refiner to make generation better. When tuning the refiner, realistically we would only need to fine tune the text encoder, not the u-net. If this ability was added, I think that would be very beneficial.

Contributor guide

No contributing guide indexed for this repository

Research direction

No specific file or test is named. Start by tracing the SDXL training scripts and existing refiner handling, then determine how text-encoder-only fine-tuning could be supported; done means the scripts can train the refiner for subject generation without fine-tuning its U-Net.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.