From 7d70d85c60c5bbe71db88fe6892a1eda769f3bea Mon Sep 17 00:00:00 2001 From: YunyaoZhou Date: Mon, 13 Jan 2025 18:26:38 +0800 Subject: [PATCH] =?UTF-8?q?feat=E2=9C=A8:=20=E6=9B=B4=E6=96=B0README?= =?UTF-8?q?=E5=92=8C=E8=AE=AD=E7=BB=83=E8=84=9A=E6=9C=AC=EF=BC=8C=E7=AE=80?= =?UTF-8?q?=E5=8C=96=E7=8E=AF=E5=A2=83=E8=AE=BE=E7=BD=AE=E5=B9=B6=E7=A7=BB?= =?UTF-8?q?=E9=99=A4=E4=B8=8D=E5=BF=85=E8=A6=81=E7=9A=84=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 21 ++------------------- src/train.py | 1 - 2 files changed, 2 insertions(+), 20 deletions(-) diff --git a/README.md b/README.md index 11b0423..188457c 100644 --- a/README.md +++ b/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 ``` diff --git a/src/train.py b/src/train.py index da5d265..491cef5 100644 --- a/src/train.py +++ b/src/train.py @@ -89,7 +89,6 @@ if __name__ == "__main__": accelerator.wait_for_everyone() model.eval() - accelerator = trainer.accelerator from torch.utils.data import DataLoader