Update core/app.py

core/app.py

@@ -24,6 +24,11 @@
 from pathlib import Path
 from typing import Optional, Tuple, Dict, Any, Callable
 
+# Mitigate CUDA fragmentation (must be set before importing torch)
+if "PYTORCH_CUDA_ALLOC_CONF" not in os.environ:
+    os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True,max_split_size_mb:128"
+
+
 # ── Logging ──────────────────────────────────────────────────────────────────
 logging.basicConfig(
     level=logging.INFO,