diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 7 |
1 files changed, 5 insertions, 2 deletions
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", |