mirror of
https://github.com/QwenLM/Qwen.git
synced 2026-05-20 16:35:47 +08:00
Update finetune.py For lower cpu memory
1.Change default device_map to "auto" device_map = "auto" 2.For lower cpu memory, add parameter at load model low_cpu_mem_usage=True,
This commit is contained in:
@@ -278,7 +278,7 @@ def train():
|
||||
|
||||
local_rank = training_args.local_rank
|
||||
|
||||
device_map = None
|
||||
device_map = "auto"
|
||||
world_size = int(os.environ.get("WORLD_SIZE", 1))
|
||||
ddp = world_size != 1
|
||||
if lora_args.q_lora:
|
||||
@@ -302,6 +302,7 @@ def train():
|
||||
config=config,
|
||||
cache_dir=training_args.cache_dir,
|
||||
device_map=device_map,
|
||||
low_cpu_mem_usage=True,
|
||||
trust_remote_code=True,
|
||||
quantization_config=GPTQConfig(
|
||||
bits=4, disable_exllama=True
|
||||
|
||||
Reference in New Issue
Block a user