From 91d440359496204c16d14c9dd072ee9aa5322dcf Mon Sep 17 00:00:00 2001 From: Ferdinand Schlatt Date: Fri, 15 Nov 2024 10:00:49 +0100 Subject: [PATCH] update inference config --- configs/trainer/inference.yaml | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/configs/trainer/inference.yaml b/configs/trainer/inference.yaml index e69de29..21bfb04 100644 --- a/configs/trainer/inference.yaml +++ b/configs/trainer/inference.yaml @@ -0,0 +1,9 @@ +seed_everything: 0 +trainer: + accelerator: auto + strategy: auto + devices: auto + num_nodes: 1 + precision: bf16-mixed + logger: false + enable_checkpointing: false