diff --git a/README.md b/README.md index dce38e5a..a04d4866 100644 --- a/README.md +++ b/README.md @@ -44,7 +44,7 @@ We provide several variants for each of the components in the unlearning pipelin ## 📌 Table of Contents - 📖 [Overview](#-overview) -- 🗃️ [Available Components](#-available-components) +- 🗃️ [Available Components](#%EF%B8%8F-available-components) - ⚡ [Quickstart](#-quickstart) - 🛠️ [Environment Setup](#-environment-setup) - 💾 [Data Setup](#-data-setup) @@ -56,7 +56,7 @@ We provide several variants for each of the components in the unlearning pipelin - ➕ [How to Add New Components](#-how-to-add-new-components) - 📚 [Further Documentation](#-further-documentation) - 🔗 [Support & Contributors](#-support--contributors) -- 📝 [Citing this work](#-citating-this-work) +- 📝 [Citing this work](#-citing-this-work) - 🤝 [Acknowledgements](#-acknowledgements) - 📄 [License](#-license) @@ -198,7 +198,7 @@ If you use OpenUnlearning in your research, please cite: --- -### 🤝 Acknowledgments +### 🤝 Acknowledgements - This repo is inspired from [LLaMA-Factory](https://github.com/hiyouga/LLaMA-Factory). - The [TOFU](https://github.com/locuslab/tofu) and [MUSE](https://github.com/jaechan-repo/muse_bench) benchmarks served as the foundation for our re-implementation. diff --git a/scripts/tofu_unlearn.sh b/scripts/tofu_unlearn.sh index a556bd1d..1794c9b6 100644 --- a/scripts/tofu_unlearn.sh +++ b/scripts/tofu_unlearn.sh @@ -13,7 +13,7 @@ trainers_experiments=( "GradAscent unlearn/tofu/default.yaml" "GradDiff unlearn/tofu/default.yaml" "NPO unlearn/tofu/default.yaml" - "DPO unlearn/tofu/default.yaml" + "DPO unlearn/tofu/idk.yaml" ) forget_retain_splits=( "forget01 retain99" diff --git a/setup_data.py b/setup_data.py index 48de0ad1..358779c3 100644 --- a/setup_data.py +++ b/setup_data.py @@ -1,8 +1,17 @@ from huggingface_hub import snapshot_download +# Setup retain model metrics snapshot_download( repo_id="open-unlearning/eval", allow_patterns="*.json", repo_type="dataset", local_dir="saves/eval", ) + +# Setup data +snapshot_download( + repo_id="open-unlearning/idk", + allow_patterns="*.jsonl", + repo_type="dataset", + local_dir="data", +) \ No newline at end of file