fix(doc): add missing low_cpu_mem_usage config to docs (#2369) [skip ci]

This commit is contained in:
NanoCode012
2025-03-05 22:01:44 +07:00
committed by GitHub
parent f18231c653
commit c8191394e9

View File

@@ -534,6 +534,8 @@ flash_attn_fuse_mlp: # Whether to fuse part of the MLP into a single operation
sdp_attention:
# Shifted-sparse attention (only llama) - https://arxiv.org/pdf/2309.12307.pdf
s2_attention:
# Optional[bool]. Whether to use low_cpu_mem_usage
low_cpu_mem_usage:
# Resume from a specific checkpoint dir
resume_from_checkpoint:
# If resume_from_checkpoint isn't set and you simply want it to start where it left off.