feat✨: 更新README和训练脚本,简化环境设置并移除不必要的代码
This commit is contained in:
parent
3b606da46f
commit
7d70d85c60
21
README.md
21
README.md
@ -3,23 +3,6 @@
|
||||
## 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
|
||||
uv sync
|
||||
uv sync --extra compile
|
||||
```
|
||||
|
@ -89,7 +89,6 @@ if __name__ == "__main__":
|
||||
accelerator.wait_for_everyone()
|
||||
|
||||
model.eval()
|
||||
accelerator = trainer.accelerator
|
||||
|
||||
from torch.utils.data import DataLoader
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user