cloud is optional

axolotl-ai-cloud · Jan 14, 2025 · 9e85bef · 9e85bef
1 parent 6349a8d
commit 9e85bef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/axolotl/cli/main.py b/src/axolotl/cli/main.py
@@ -63,7 +63,7 @@ def preprocess(config: str, cloud: Optional[str] = None, **kwargs) -> None:
 @add_options_from_dataclass(TrainerCliArgs)
 @add_options_from_config(AxolotlInputConfig)
 @filter_none_kwargs
-def train(config: str, accelerate: bool, cloud: Optional[str], **kwargs) -> None:
+def train(config: str, accelerate: bool, cloud: Optional[str] = None, **kwargs) -> None:
     """
     Train or fine-tune a model.