search

ZichengDuan / TheChosenOne

Unofficial implementation of the paper "The Chosen One: Consistent Characters in Text-to-Image Diffusion Models"
https://arxiv.org/abs/2311.10093
237 stars 22 forks source link

GPU requirements #5

Closed NirDiamant closed 9 months ago

NirDiamant commented 9 months ago

what are the minimal GPU requirements for training the model? currently, I'm running out of CUDA memory, using a batch size of 1, using a 16GB RAM GPU

ZichengDuan commented 9 months ago

Hi, thanks for your interest, it took me about ~23GB resources to train, I am using a single RTX 4090.