From fc6e8636014dbaf882a2fa5685497dc225ee8e29 Mon Sep 17 00:00:00 2001
From: pks <pks@pks.rocks>
Date: Sat, 29 Nov 2025 23:56:12 +0100
Subject: WIP

---
 pyproject.toml | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'pyproject.toml')

diff --git a/pyproject.toml b/pyproject.toml
index bbf1288..c650ffa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,16 +1,18 @@
 [project]
-name = "gemma3_4b"
+name = "CaptionTranslation"
 version = "0.1.0"
-description = "Add your description here"
+description = "Finetuning gemma3 towards translating captions"
 requires-python = ">=3.13"
 dependencies = [
     "absl-py>=2.3.1",
     "accelerate>=1.11.0",
+    "adamw-bf16>=0.0.3",
     "bitsandbytes>=0.48.2",
     "datasets>=4.4.1",
     "diskcache>=5.6.3",
     "huggingface-hub>=0.36.0",
     "ipdb>=0.13.13",
+    "jsonargparse[signatures]>=4.35.0",
     "ninja>=1.13.0",
     "openai>=2.8.0",
     "packaging>=25.0",
@@ -18,6 +20,7 @@ dependencies = [
     "pillow>=12.0.0",
     "sacrebleu>=2.5.1",
     "termcolor>=3.2.0",
+    "torchao>=0.14.1",
     "torchvision>=0.24.1",
     "transformers>=4.57.1",
     "trl>=0.25.1",
-- 
cgit v1.2.3