Nanobit commited on
Commit
2824423
1 Parent(s): cb18856

Add warning for `trust_remote_code`

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/validation.py +3 -0
src/axolotl/utils/validation.py CHANGED
@@ -18,6 +18,9 @@ def validate_config(cfg):
18
 
19
  if not cfg.load_in_8bit and cfg.adapter == "lora":
20
  logging.warning("We recommend setting `load_in_8bit: true` for LORA finetuning")
 
 
 
21
 
22
  # TODO
23
  # MPT 7b
 
18
 
19
  if not cfg.load_in_8bit and cfg.adapter == "lora":
20
  logging.warning("We recommend setting `load_in_8bit: true` for LORA finetuning")
21
+
22
+ if cfg.trust_remote_code:
23
+ logging.warning("`trust_remote_code` is set to true. Please make sure that you reviewed the remote code/model.")
24
 
25
  # TODO
26
  # MPT 7b