summaryrefslogtreecommitdiff
path: root/pyproject.toml
diff options
context:
space:
mode:
Diffstat (limited to 'pyproject.toml')
-rw-r--r--pyproject.toml25
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" }