feat: 更新README和训练脚本,简化环境设置并移除不必要的代码

This commit is contained in:
YunyaoZhou 2025-01-13 18:26:38 +08:00
parent 3b606da46f
commit 7d70d85c60
Signed by: shujakuin
GPG Key ID: 418C3CA28E350CCF
2 changed files with 2 additions and 20 deletions

View File

@ -3,23 +3,6 @@
## Requirements ## Requirements
```bash ```bash
uv venv --python 3.11.7 uv sync
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 sync --extra compile
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

@ -89,7 +89,6 @@ if __name__ == "__main__":
accelerator.wait_for_everyone() accelerator.wait_for_everyone()
model.eval() model.eval()
accelerator = trainer.accelerator
from torch.utils.data import DataLoader from torch.utils.data import DataLoader