search

ygtxr1997 / CelebBasis

Official Implementation of 'Inserting Anybody in Diffusion Models via Celeb Basis'
https://celeb-basis.github.io/
MIT License
252 stars 7 forks source link

What are the CPU RAM requirements? VRAM requirements? #6

Closed mjdi closed 10 months ago

mjdi commented 1 year ago

I've been running it so far via Windows Subsystem for Linux 2, and the 01_start_train.sh bash script was crashing at the default 16 GB of RAM for my 32 GB Ubuntu install. I had to increase it to 26 GB (Windows 11 needs about 4 GB to operate) via the .wslconfig or it was getting killed due to lack of memory https://fizzylogic.nl/2023/01/05/how-to-configure-memory-limits-in-wsl2 Task_Manager_00_37_36

I think there is an option to use the gpu to load the models directly, rather than copy them into RAM first, but I suppose this wouldn't be good for low VRAM GPUs (I have a 3090).

mjdi commented 1 year ago

It seemed to use about 21GB of VRAM when training