Hello everyone. I am Dr. Furkan Gözükara. PhD Computer Engineer. SECourses is a dedicated YouTube channel for the following topics : Tech, AI, News, Science, Robotics, Singularity, ComfyUI, SwarmUI, ML, Artificial Intelligence, Humanoid Robots, Wan 2.2, FLUX, Krea, Qwen Image, VLMs, Stable Diffusion
torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB. GPU 0 has a total capacity of 14.74 GiB of which 14.12 MiB is free. Process 16031 has 14.72 GiB memory in use. Of the allocated memory 14.26 GiB is allocated by PyTorch, and 344.88 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation.
v25 worked. Thanks for the help. But I couldn't use both the gpus, when I've selected multi_gpu and Number of processes=2 it is throwing an error. I ran it with one gpu.
I saved 3 models during the training steps (1 at every 601 steps). apart from these 3, there are 6 models in the same folder namely Kaggle_SDXL_Base_DreamBooth-000001.safetensors. what are these models? can these be used to extract lora from dreambooth models(Kaggle_SDXL_Base_DreamBooth-step00001803.safetensors)
I'm testing the model results, I see in some generated photos the background and clothes of training images are coming up. If I include captions while training, will this issue be resolved?
Hi Dr Furkan, Shaun here. I'm planning to fine-tune an SDXL model checkpoint on a large number of images, hoping to improve its base performance. Some examples are like JuggernautXL , RealVis XL etc. Were they trained using dreambooth as well? Is dreambooth the best method to train a general model? As a lot of the examples of dreambooth seem to be only about training on a specific subject.
I'm using 13 images now with 4 unique backgrounds and clothes, can you tell me how many images are needed? how many steps should I train? config file has train batch size as 2, should I try train batch size =1. Does it help if I remove the background in the train images?