7 Commits (07de5117668ee333bea70a3b0d3bb84a9a557d05)

Author SHA1 Message Date
梦典 f3d7c69be2
Update finetune.py
Just lora finetuing need low cpu memory usage
1 year ago
梦典 67b3f949b6
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,
1 year ago
Junyang Lin 00dceebbaa
Update finetune.py 2 years ago
Junyang Lin f7681f9e77
Update finetune.py 2 years ago
Junyang Lin 75cc160f74
Update finetune.py 2 years ago
JustinLin610 b5fad3d561 fix single-gpu qlora, and add profiling 2 years ago
JustinLin610 af22d5e0ce add finetuning 2 years ago