Improved gemma training

2026-01-11 16:25:40 +01:00
parent 289a52d6ce
commit aee72e65ec
2 changed files with 6 additions and 5 deletions
--- a/tools/gemma_train/pyproject.toml
+++ b/tools/gemma_train/pyproject.toml
@@ -12,6 +12,7 @@ dependencies = [
    "bitsandbytes>=0.45.0",
    "datasets==3.3.2",
    "evaluate==0.4.3",
+    "kernels>=0.11.1",
    "llm_engine_utils @ file:///root/sia/lib/llm_engine_utils",
    "peft==0.13.2",
    "python-dotenv>=1.0.0",