Gemma training script

This commit is contained in:
Niels Geens
2025-05-20 20:46:16 +02:00
parent d4a4902b94
commit f2c70cd05d
14 changed files with 394 additions and 207 deletions

View File

@@ -0,0 +1,21 @@
[build-system]
requires = ["setuptools>=42", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "gemma_train"
version = "0.1.0"
requires-python = ">=3.8"
dependencies = [
"accelerate>=0.26.0",
"bitsandbytes>=0.45.0",
"llama-cpp-scripts @ file:///root/sia/modules/llama.cpp",
"llm_engine_utils @ file:///root/sia/lib/llm_engine_utils",
"trl>=0.17.0",
"peft>=0.15.0",
"python-dotenv>=1.0.0",
]
[project.scripts]
gemma_train = "gemma_train.__main__:main"