Renamed mistral tool to mistral_api
This commit is contained in:
27
tools/mistral_local_train/pyproject.toml
Normal file
27
tools/mistral_local_train/pyproject.toml
Normal file
@@ -0,0 +1,27 @@
|
||||
[build-system]
|
||||
requires = ["setuptools>=42", "wheel"]
|
||||
build-backend = "setuptools.build_meta"
|
||||
|
||||
[project]
|
||||
name = "mistral_local_train"
|
||||
version = "0.1.0"
|
||||
requires-python = ">=3.8"
|
||||
|
||||
dependencies = [
|
||||
"accelerate==1.2.0",
|
||||
"bitsandbytes>=0.45.0",
|
||||
"datasets==3.3.2",
|
||||
"evaluate==0.4.3",
|
||||
"kernels>=0.11.1",
|
||||
"llm_engine_utils @ file:///root/sia/lib/llm_engine_utils",
|
||||
"mistral-common>=1.8.6",
|
||||
"peft==0.13.2",
|
||||
"python-dotenv>=1.0.0",
|
||||
"sentencepiece>=0.2.0",
|
||||
"torch==2.4.1",
|
||||
"transformers>=5.0.0rc0",
|
||||
"trl>=0.17.0",
|
||||
]
|
||||
|
||||
[project.scripts]
|
||||
mistral_local_train = "mistral_local_train.__main__:main"
|
||||
Reference in New Issue
Block a user