kohya-ss / kohya-ss/sd-scripts

sdxl lora training seems to consume much more gpu memory than normal sdxl training does?

Open
#816 1 comment 0 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.