Skip to content

Commit 334e219

Browse files
committed
update README.md
1 parent 7e4346f commit 334e219

File tree

3 files changed

+43
-2
lines changed

3 files changed

+43
-2
lines changed

README.md

+1-2
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22

33
<p align="center">
44
<a href="https://arxiv.org/abs/2502.11190">📄arXiv</a> •
5-
<a href="">🤗HFPaper</a> •
6-
<a href="">🎧NotebookLM Audio</a>
5+
<a href="https://huggingface.co/papers/2502.11190">🤗HFPaper</a>
76
</p>
87

98
This repository provides the official PyTorch implementation of the following paper:

semeval25/README.md

+24
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# SemEval Unlearning
2+
This folder contains the solution developed by ZJUKLAB for the [SemEval 2025 Task 4](https://llmunlearningsemeval2025.github.io/) competition.
3+
4+
## Installation
5+
6+
7+
```bash
8+
conda create -n semeval_unlearn python=3.12
9+
conda activate semeval_unlearn
10+
pip install -r requirements.txt
11+
```
12+
13+
### Script Arguments
14+
15+
- `--forget_dataset`: Specifies the dataset to forget (must be a valid dataset path or identifier).
16+
- `--retain_dataset`: Specifies the dataset to retain.
17+
- `--model_path`: Path to the pre-trained model.
18+
- `--output_dir`: Directory where results and logs will be saved.
19+
20+
### Run the Script:
21+
22+
```bash
23+
torchrun --nproc_per_node=1 --master_port=29500 unlearn-merging.py --forget_dataset /path/to/forget_data --retain_dataset /path/to/retain_data --model_path /path/to/model --output_dir /path/to/output
24+
```

semeval25/requirements.txt

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
datasets
2+
accelerate
3+
deepspeed
4+
evaluate
5+
matplotlib
6+
hydra-core
7+
omegaconf
8+
peft
9+
rouge_score
10+
tqdm
11+
matplotlib
12+
einops
13+
packaging
14+
bitsandbytes
15+
scipy
16+
ninja
17+
vllm
18+
wandb

0 commit comments

Comments
 (0)