From 9bb149acdbf54e10a260f346ebca315871e9a2a2 Mon Sep 17 00:00:00 2001
From: "stephane.david" <thedarty@gmail.com>
Date: Fri, 19 Jun 2026 16:16:25 +0200
Subject: [PATCH] Add PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True env

---
 docker-compose.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker-compose.yml b/docker-compose.yml
index 87e8871..6eae9e4 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -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"