From 2824423d10b24fc22969c4342e1bd2ca8f6be6da Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Sun, 28 May 2023 00:46:56 +0900 Subject: [PATCH] Add warning for `trust_remote_code` --- src/axolotl/utils/validation.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/axolotl/utils/validation.py b/src/axolotl/utils/validation.py index f85cc37f7..bd759c3b7 100644 --- a/src/axolotl/utils/validation.py +++ b/src/axolotl/utils/validation.py @@ -18,6 +18,9 @@ def validate_config(cfg): if not cfg.load_in_8bit and cfg.adapter == "lora": logging.warning("We recommend setting `load_in_8bit: true` for LORA finetuning") + + if cfg.trust_remote_code: + logging.warning("`trust_remote_code` is set to true. Please make sure that you reviewed the remote code/model.") # TODO # MPT 7b