fix(ft): set report_to none to avoid wandb setup

Signed-off-by: aarnphm-ec2-dev <[email protected]>
GutZuFusss · Jul 21, 2023 · beb8c2b · beb8c2b
1 parent 8a42832
commit beb8c2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/openllm/playground/llama2_qlora.py b/src/openllm/playground/llama2_qlora.py
@@ -162,7 +162,7 @@ class TrainingArguments:
     learning_rate: float = dataclasses.field(default=5e-5)
     num_train_epochs: int = dataclasses.field(default=3)
     logging_steps: int = dataclasses.field(default=1)
-    logging_strategy: str = dataclasses.field(default="steps")
+    report_to: str = dataclasses.field(default="none")
     output_dir: str = dataclasses.field(default=os.path.join(os.getcwd(), "outputs", "llama"))
     save_strategy: str = dataclasses.field(default="no")