diff --git a/README-zh.md b/README-zh.md index b50d541..62fdb75 100644 --- a/README-zh.md +++ b/README-zh.md @@ -88,7 +88,12 @@ DataFlex 与 LLaMA-Factory 无缝集成,为研究人员和开发者提供更 git clone https://github.com/OpenDCAI/DataFlex.git cd DataFlex pip install -e . + +# 在 Python 3.10 环境中,请安装 v0.9.3 以确保兼容性 pip install llamafactory==0.9.3 + +# 在 Python 3.11+ 环境中,推荐安装最新的 v0.9.4 +pip install llamafactory==0.9.4 ``` 启动命令与 [LLaMA-Factory](https://github.com/hiyouga/LLaMA-Factory) 类似。 diff --git a/README.md b/README.md index a3613a1..edf7a73 100644 --- a/README.md +++ b/README.md @@ -90,7 +90,12 @@ Please use the following commands for environment setup and installation👇 git clone https://github.com/OpenDCAI/DataFlex.git cd DataFlex pip install -e . + +# In a Python 3.10 environment pip install llamafactory==0.9.3 + +# In a Python 3.11+ environment +pip install llamafactory==0.9.4 ``` The launch command is similar to [LLaMA-Factory](https://github.com/hiyouga/LLaMA-Factory). diff --git a/requirements.txt b/requirements.txt index a284568..26dec6d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,17 @@ -transformers>=4.41.2,<=4.50.0,!=4.46.*,!=4.47.*,!=4.48.*;python_version<'3.10' -transformers>=4.41.2,<=4.50.0,!=4.46.*,!=4.47.*,!=4.48.0;python_version>='3.10' +torch>=2.4.0,<=2.10.0 +torchvision>=0.19.0,<=0.20.0 +torchaudio>=2.4.0,<=2.5.0 +transformers>=4.51.0,<4.55.0; python_version >= '3.11' +transformers>=4.41.2,<=4.50.0; python_version < '3.11' +accelerate>=1.3.0,<1.12.0 +trl>=0.18.0,<0.25.0 +tyro==0.8.14 +numpy>=1.24.0,<2.0.0 +peft>=0.15.2 datasets>=2.16.0,<=3.2.0 -accelerate>=0.34.0,<=1.2.1 -peft>=0.14.0,<=0.15.2 -trl>=0.8.6,<=0.9.6 -tokenizers>=0.19.0,<=0.21.0 +tokenizers>=0.21.0 gradio>=4.38.0,<=5.12.0 -pandas>=2.0.0 +pandas>=2.0.0,<3.0.0 scipy einops sentencepiece @@ -20,10 +25,7 @@ matplotlib>=3.7.0 fire packaging pyyaml -numpy<2.0.0 av librosa -tyro<0.9.0 omegaconf -traker -#faiss +traker \ No newline at end of file