autotrain-advanced
autotrain-advanced copied to clipboard
Liger kernel
This PR adds support for Liger Kernel to enable lower training times and reduce memory usage.
It implements:
- Validating
liger-kernel
is available. - Adds training
liger-kernel
parameter to llm finetuning tasks. - Switching to
liger-kernel
's model loader then installed, enabled by the user and the LLM's type is supported byliger-kernel
.