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,main
parent
78352b5a79
commit
67b3f949b6
Loading…
Reference in New Issue