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