更新README.md以添加项目要求和代码结构,修改requirements.txt以移除不必要的依赖项

This commit is contained in:
YunyaoZhou 2025-01-02 21:50:39 +08:00
parent 2062f90e5d
commit f230886c3f
Signed by: shujakuin
GPG Key ID: 418C3CA28E350CCF
2 changed files with 25 additions and 6 deletions

View File

@ -1 +1,25 @@
# CL-LMM # CL-LMM
## Requirements
```bash
uv venv --python 3.11.7
uv pip install torch==2.5.1+cu124 torchvision==0.20.1+cu124 torchaudio==2.5.1+cu124 --index-url https://download.pytorch.org/whl/cu124
uv pip install -r requirements.txt
```
## Code Structure
```bash
src
├── collatefn_library
├── configs accelerator configs
├── dataset_library
├── evaluation.py
├── evaluation.sh evaluation bash(TODO:update)
├── peft_library
├── todo.md
├── train.py
├── train.sh train bash
└── utils
```

View File

@ -1,11 +1,6 @@
accelerate==1.2.1 accelerate==1.2.1
deepspeed==0.16.2 deepspeed==0.16.2
evaluate==0.4.3 evaluate==0.4.3
networkx==3.2.1
ninja==1.11.1.3
numpy==1.26.3
packaging==24.2
pandas==2.2.3
peft==0.14.0 peft==0.14.0
pillow==10.2.0 pillow==10.2.0
torch==2.5.1+cu124 torch==2.5.1+cu124