[BUG] My Docker container frequently restarts

[SilentTE]

Model
Which model are you using?
lama

Describe the bug
A clear and concise description of what the bug is.
CUDA error: an illegal memory access was encounteredCUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.For debugging consider passing CUDA_LAUNCH_BLOCKING=1.Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.
Screenshots
If applicable, add screenshots to help explain your problem.

System Info
Software version used

• iopaint:
• pytorch:
• CUDA:12.0