fix: Include kernels (allows MXFP4 to be loaded in MXFP4 instead of upcasting) (#107)

Co-authored-by: Andrew Patrikalakis <anrp@tri.global>
2026-01-16 12:00:24 +00:00
parent 632b1da622
commit 85a6ec5ecb
2 changed files with 20 additions and 2 deletions
@@ -27,12 +27,13 @@ dependencies = [
    "datasets>=4.0.0",
    "hf-transfer>=0.1.9",
    "huggingface-hub>=0.34.4",
+    "kernels>=0.11.7",
    "optuna>=4.5.0",
    "peft>=0.14.0",
    "pydantic-settings>=2.10.1",
    "questionary>=2.1.1",
    "rich>=14.1.0",
-    "transformers>=4.55.2",
+    "transformers>=4.57.3",
 ]

 [project.optional-dependencies]