diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..1f21c48 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,25 @@ +[project] +name = "gemma3_4b" +version = "0.1.0" +description = "Add your description here" +requires-python = ">=3.13" +dependencies = [ + "absl-py>=2.3.1", + "accelerate>=1.11.0", + "bitsandbytes>=0.48.2", + "datasets>=4.4.1", + "diskcache>=5.6.3", + "ipdb>=0.13.13", + "ninja>=1.13.0", + "openai>=2.8.0", + "packaging>=25.0", + "peft>=0.18.0", + "pillow>=12.0.0", + "termcolor>=3.2.0", + "torchvision>=0.24.1", + "transformers>=4.57.1", + "trl>=0.25.1", +] + +[tool.uv.sources] +trl = { git = "https://github.com/huggingface/trl.git" } |
