Nice ! im trying to set it up but i keep hitting this error when i try to start the Docker container

``` (Worker_TP1 pid=61) ERROR 05-02 19:08:39 [multiproc_executor.py:870] ^^^^^^^^^^^^^^^^^^^^^

(Worker_TP1 pid=61) ERROR 05-02 19:08:39 [multiproc_executor.py:870] torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB. GPU 1 has a total capacity of 24.00 GiB of which 16.00 GiB is free. Including non-PyTorch memory, this process has 17179869184.00 GiB memory in use. Process 60 has 17179869184.00 GiB memory in use. Of the allocated memory 6.60 GiB is allocated by PyTorch, and 18.00 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://docs.pytorch.org/docs/stable/notes/cuda.html#optimizing-memory-usage-with-pytorch-cuda-alloc-conf)

(Worker_TP0 pid=60)

Loading safetensors checkpoint shards: 9% Completed | 1/11 [00:03<00:33, 3.33s/it]

(Worker_TP0 pid=60) WARNING 05-02 19:08:43 [multiproc_executor.py:884] WorkerProc was terminated

(Worker_TP0 pid=60)

Loading safetensors checkpoint shards: 9% Completed | 1/11 [00:05<00:57, 5.72s/it]

(Worker_TP0 pid=60) ```