training code

2025-09-29 16:13:18 +09:00
parent d3793890a7
commit 1b445444cf
7 changed files with 406 additions and 22 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,5 +10,8 @@ dependencies = [
    "ruff>=0.13.2",
    "safetensors>=0.6.2",
    "torch>=2.8.0",
+    "torchvision>=0.23.0",
+    "tqdm>=4.67.1",
    "transformers>=4.56.2",
+    "wandb>=0.22.0",
 ]