feat(truss: add a new field to CheckpointingConfig (#2026)

aghilann · Aghilan Nathan · web-flow · commit 86df951c6390 · 2025-11-06T11:22:02.000-07:00
Co-authored-by: Aghilan Nathan &lt;aghilan.nathan@baseten.co&gt;
diff --git a/truss-train/truss_train/definitions.py b/truss-train/truss_train/definitions.py
@@ -99,6 +99,7 @@ class LoadCheckpointConfig(custom_types.SafeModelNoExtra):
 class CheckpointingConfig(custom_types.SafeModelNoExtra):
     enabled: bool = False
     checkpoint_path: Optional[str] = None
+    volume_size_gib: Optional[int] = None
 
 
 class CacheConfig(custom_types.SafeModelNoExtra):
