From 930472b7c7f4c2b5c550d5e26c811986f399c780 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Wed, 14 May 2025 16:56:34 +0700 Subject: [PATCH] chore: add missing config to doc --- docs/config.qmd | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/config.qmd b/docs/config.qmd index f03c9da06..2e0e25987 100644 --- a/docs/config.qmd +++ b/docs/config.qmd @@ -483,6 +483,7 @@ output_dir: ./completed-model # setting to `auto` will enable torch compile when torch>=2.5.1 torch_compile: # Optional[Union[Literal["auto"], bool]] torch_compile_backend: # Optional[str] +torch_compile_mode: # 'default' | 'reduce-overhead' | 'max-autotune' # Training hyperparameters