llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixes for 12B

Browse Source
This commit is contained in:
Niels Geens
2025-05-22 18:51:07 +00:00
parent 19dd9fa2ee
commit c22917e3c9
4 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tools/gemma_train/pyproject.toml
View File

@@ -8,17 +8,17 @@ version = "0.1.0"
requires-python = ">=3.8"
dependencies = [
"accelerate==1.4.0",
"bitsandbytes==0.45.3",
"accelerate==1.2.0",
"bitsandbytes>=0.45.0",
"datasets==3.3.2",
"evaluate==0.4.3",
"llm_engine_utils @ file:///root/sia/lib/llm_engine_utils",
"peft==0.14.0",
"peft==0.13.2",
"python-dotenv>=1.0.0",
"sentencepiece>=0.2.0",
"torch>=2.4.0",
"transformers>=4.51.3",
"trl==0.15.2",
"torch==2.4.1",
"transformers==4.51.3",
"trl>=0.17.0",
]
[project.scripts]