From 981c89b2a95676a4f98e94218c192c095bed5364 Mon Sep 17 00:00:00 2001 From: yangapku Date: Tue, 14 Nov 2023 18:22:06 +0800 Subject: [PATCH] update finetune.py --- finetune.py | 1 - 1 file changed, 1 deletion(-) diff --git a/finetune.py b/finetune.py index 04ad907..a756556 100644 --- a/finetune.py +++ b/finetune.py @@ -302,7 +302,6 @@ def train(): config=config, cache_dir=training_args.cache_dir, device_map=device_map, - low_cpu_mem_usage=True if training_args.use_lora and not lora_args.q_lora else False, trust_remote_code=True, quantization_config=GPTQConfig( bits=4, disable_exllama=True