diff --git a/docker-compose.yml b/docker-compose.yml
index b195d0f3..2017c8e7 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -8,7 +8,7 @@ services:
     environment:
       - OMP_NUM_THREADS=128
       - CUDA_LAUNCH_BLOCKING=0
-      - TORCH_SHOW_CPP_STACKTRACES=0
+      - PYTHONUNBUFFERED=1
     command:
       - /model
       - --trust-remote-code