Add PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True env

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