From 40e8aaa05b925a2ac7eddf1def6e970969232130 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Wed, 24 Sep 2025 12:05:19 +0200
Subject: [PATCH] fix

---
 src/lerobot/configs/train.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/lerobot/configs/train.py b/src/lerobot/configs/train.py
index 60a4d81d5..f2d07cd7f 100644
--- a/src/lerobot/configs/train.py
+++ b/src/lerobot/configs/train.py
@@ -63,6 +63,10 @@ class TrainPipelineConfig(HubMixin):
     scheduler: LRSchedulerConfig | None = None
     eval: EvalConfig = field(default_factory=EvalConfig)
     wandb: WandBConfig = field(default_factory=WandBConfig)
+    # Accelerate configuration for multi-GPU training
+    use_accelerate: bool = False
+    gradient_accumulation_steps: int = 1
+    mixed_precision: str = "no"  # Options: "no", "fp16", "bf16"
 
     def __post_init__(self):
         self.checkpoint_path = None