Running Dreambooth in Stable Dif
2023-03-17 本文已影响0人
秦汉邮侠
Error occurs when Running Dreambooth in Stable Diffusion
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 146.00 MiB (GPU 0; 4.00 GiB total capacity; 3.38 GiB already allocated; 0 bytes free; 3.43 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
Traceback (most recent call last):
File "E:\github\stable-diffusion-webui\extensions\sd_dreambooth_extension\dreambooth\ui_functions.py", line 727, in start_training
result = main(class_gen_method=class_gen_method)
File "E:\github\stable-diffusion-webui\extensions\sd_dreambooth_extension\dreambooth\train_dreambooth.py", line 1451, in main
return inner_loop()
File "E:\github\stable-diffusion-webui\extensions\sd_dreambooth_extension\dreambooth\memory.py", line 117, in decorator
raise RuntimeError("No executable batch size found, reached zero.")
RuntimeError: No executable batch size found, reached zero.
Reffere Running Dreambooth in Stable Diffusion with Low VRAMnce
Link:https://jeskin.net/blog/dreambooth-low-vram
I don't konw whether this solution is worked