[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

[github-actions]

This issue is stale because it has been open for 30 days with no activity.

[github-actions]

This issue was closed because it has been inactive for 14 days since being marked as stale.