ProTrain: Efficient LLM Training via Memory-Aware Techniques
Paper ⢠2406.08334 ⢠Published
None defined yet.
hf-mem v0.4.1 now also estimates KV cache memory requirements for any context length and batch size with the --experimental flag!uvx hf-mem --model-id ... --experimental will automatically pull the required information from the Hugging Face Hub to include the KV cache estimation, when applicable.--max-model-len, --batch-size and --kv-cache-dtype arguments (Ć la vLLM) manually if preferred.