From d485a083938e995979d1a566bb6a2f876075c667 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Sat, 16 Mar 2024 21:10:38 +0900 Subject: [PATCH] chore(script): remove redundant setting (#1411) --- src/axolotl/train.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/axolotl/train.py b/src/axolotl/train.py index 5c4eaf3d5..d2fc75261 100644 --- a/src/axolotl/train.py +++ b/src/axolotl/train.py @@ -110,9 +110,6 @@ def train( total_num_steps, ) - if hasattr(model, "config"): - model.config.use_cache = False - # go ahead and presave, so we have the adapter config available to inspect if peft_config: LOG.info(f"Pre-saving adapter config to {cfg.output_dir}")