Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions README-zh.md
Original file line number Diff line number Diff line change
Expand Up @@ -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) 类似。
Expand Down
5 changes: 5 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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).
Expand Down
24 changes: 13 additions & 11 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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
Loading