Add PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True env

2026-06-19 16:23:43 +02:00
parent 9bb149acdb
commit ebbe43deb6
1 changed files with 1 additions and 1 deletions
@@ -5,7 +5,7 @@ services:
    network_mode: bridge
    environment:
      - HF_TOKEN=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
-      - PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
+      - PYTORCH_ALLOC_CONF=expandable_segments:True,throw_on_cudamalloc_oom:True
    ipc: host
    ports:
      - "8000:8000"