CUDA error 2: out of memory

[ParanoidPY]

When I run the inference demo code: "python inference/sample_shapenet.py none --category car --total_len 20 --batch_len 4 --ema --use_ddim --ddim_step 100 --extract_mesh".
it appears an error that shows "CUDA error 2: out of memory (/media/py/HDD1/3D/DiffusionModel/XCube/openvdb/fvdb/../nanovdb/nanovdb/cuda/GridHandle.cuh:83)".
My GPU is 4090 with 24g cuda memory. I want to know the least memory volume to run the code.
THANKS.

[xrenaa]

Hi, you could try with batch_len 1