huggingface / huggingface/diffusers

DreamBooth LoRA SDXL OOM issue at the last inference (8xRTX3090 24G)

Open
#10,829 1 comment 0 reactions 0 assignees View on GitHub
stale
Dominant language
Python
Stars
34.5k
Forks
7.3k
Avg merge
3d 3h
Merged PRs (30d)
91

Description

Hi, I'm encountering a CUDA out of memory error during the final inference step when running the 3D Icon example from the [advanced diffusion training](https://github.com/huggingface/diffusers/tree/main/examples/advanced_diffusion_training#3d-icon-example).

Script: I used the exact same script provided in the example.
Issue: Training epochs complete successfully without any OOM errors.
But the error occurs during the last inference step at [L2416](https://github.com/huggingface/diffusers/blob/6fe05b9b93593bca41afac79b32b7a23526b0e96/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py#L2416).

Tried Solutions:
Reduced --num_validation_images to 1, but the OOM still happens.

Environment:
GPUs: 8 x RTX3090 24GB
PyTorch version: 2.6.0
CUDA version: 12.4
diffusers version: 0.33.0.dev0

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.