From f230886c3fad9fbe7fecec5ae8a63c8e97aec62d Mon Sep 17 00:00:00 2001 From: YunyaoZhou Date: Thu, 2 Jan 2025 21:50:39 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0README.md=E4=BB=A5=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0=E9=A1=B9=E7=9B=AE=E8=A6=81=E6=B1=82=E5=92=8C=E4=BB=A3?= =?UTF-8?q?=E7=A0=81=E7=BB=93=E6=9E=84=EF=BC=8C=E4=BF=AE=E6=94=B9requireme?= =?UTF-8?q?nts.txt=E4=BB=A5=E7=A7=BB=E9=99=A4=E4=B8=8D=E5=BF=85=E8=A6=81?= =?UTF-8?q?=E7=9A=84=E4=BE=9D=E8=B5=96=E9=A1=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 26 +++++++++++++++++++++++++- requirements.txt | 5 ----- 2 files changed, 25 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index 3cb81ea..11b0423 100644 --- a/README.md +++ b/README.md @@ -1 +1,25 @@ -# CL-LMM \ No newline at end of file +# 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 +``` diff --git a/requirements.txt b/requirements.txt index 245f9a1..c09c604 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,11 +1,6 @@ accelerate==1.2.1 deepspeed==0.16.2 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 pillow==10.2.0 torch==2.5.1+cu124