update finetune.py

2026-05-20 16:35:47 +08:00 · 2023-11-14 18:22:06 +08:00
parent 91675cbcd4
commit 981c89b2a9
1 changed files with 0 additions and 1 deletions
--- 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