v2; a lot of hacks

2025-10-15 18:02:10 +09:00
parent e51017897d
commit 6ab33ceb83
7 changed files with 586 additions and 199 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,8 @@ dependencies = [
    "tqdm>=4.67.1",
    "transformers>=4.56.2",
    "wandb[media]>=0.22.0",
-    "flash-attn"
+    "flash-attn",
+    "lovely-tensors>=0.1.19",
 ]

 [tool.uv.extra-build-dependencies]