ToTheBeginning / PuLID

[NeurIPS 2024] Official code for PuLID: Pure and Lightning ID Customization via Contrastive Alignment
Apache License 2.0
2.12k stars 137 forks source link

out of CUDA memery in 3060 12GB GPU #13

Open inspire-boy opened 5 months ago

inspire-boy commented 5 months ago

This is my errors: torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 22.00 MiB (GPU 0; 12.00 GiB total capacity; 10.89 GiB already allocated; 0 bytes free; 11.31 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

PLatform:WIN10 / 3060 12G GPU

When I change Num samples to 1,There's error too. What should I Change In code? Thanks!