Running Dreambooth in Stable Diffusion with Low VRAM

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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容