diff --git a/docker-compose.yml b/docker-compose.yml index 947481c5..fdd0bd98 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -10,6 +10,7 @@ services: - CUDA_LAUNCH_BLOCKING=0 - PYTHONUNBUFFERED=1 - VLLM_RPC_TIMEOUT_MS=600000 + - CLAWMINE_NAN_CHECK=1 command: - /model - --trust-remote-code