tiny fix

xin-li-67 · Mar 14, 2024 · 6ebde4f · 6ebde4f
1 parent 3b4a59b
commit 6ebde4f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/llmtuner/train/tuner.py b/src/llmtuner/train/tuner.py
@@ -42,7 +42,6 @@ def run_exp(args: Optional[Dict[str, Any]] = None, callbacks: Optional[List["Tra
 
 def export_model(args: Optional[Dict[str, Any]] = None):
     model_args, data_args, finetuning_args, _ = get_infer_args(args)
-    model_args.low_cpu_mem_usage = False
 
     if model_args.export_dir is None:
         raise ValueError("Please specify `export_dir` to save model.")