feat: add model reload endpoint and forge CLI

2026-04-13 15:21:06 -04:00
parent e919d2a8e2
commit 47ac2f36e0
5 changed files with 412 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,6 +26,16 @@ dev = [
    "httpx>=0.27.0",
    "respx>=0.21.0",
 ]
+train = [
+    "unsloth>=2024.1.0",
+    "torch>=2.1.0",
+    "transformers>=4.36.0",
+    "datasets>=2.14.0",
+    "peft>=0.7.0",
+    "accelerate>=0.25.0",
+    "bitsandbytes>=0.41.0",
+    "trl>=0.7.0",
+]

 [tool.hatchling]
 packages = ["src/companion"]