kohya-ss / kohya-ss/sd-scripts
sdxl lora training seems to consume much more gpu memory than normal sdxl training does?
- Dominant language
- Python
- Stars
- 7.2k
- Forks
- 1.2k
- Avg merge
- 11m
- Merged PRs (30d)
- 2
Description
Specifically, sdxl_train v.s. sdxl_train_network
I have compared the trainable params, the are the same, and the training params are the same.
As a result, batch size 10 --> 4 otherwise an gpu memory error will occur.
So a lora module[64dim 32 alpha] consumes 6 batch size.
Any one give me some hints, thanks!
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by comparing the sdxl_train and sdxl_train_network entry points with the reported matching trainable and training parameters. Reproduce the GPU-memory difference using the stated batch-size change and 64-dimension, 32-alpha LoRA settings; done means identifying the cause and preventing the unexpected memory increase or documenting the required memory behavior.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 32/100