[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