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