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

上一篇下一篇

猜你喜欢

热点阅读