Update README.md

README.md

@@ -1,13 +1,34 @@
 ---
 license: mit
+datasets:
+- locuslab/TOFU
+language:
+- en
+base_model:
+- NousResearch/Llama-2-7b-chat-hf
+pipeline_tag: text-generation
+library_name: transformers
+tags:
+- unlearn
+- machine-unlearning
+- llm-unlearning
+- data-privacy
+- large-language-models
+- trustworthy-ai
+- trustworthy-machine-learning
+- language-model
 ---
 
-# LLaMA-2-chat 7B fine tunes on TOFU (Origin Model)
+# Origin Model on Task "TOFU"
 
 ## Model Details
+- **Training**:
+  - **Task**: [🤗datasets/locuslab/TOFU](https://huggingface.co/datasets/locuslab/TOFU)
+  - **Method**: Fine tune
+- **Base Model**: [[🤗NousResearch/Llama-2-7b-chat-hf](https://huggingface.co/NousResearch/Llama-2-7b-chat-hf)
+- **Code Base**: [github.com/OPTML-Group/Unlearn-Simple](https://github.com/OPTML-Group/Unlearn-Simple)
+- **Research Paper**: ["Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning"](https://arxiv.org/abs/2410.07163)
 
-- **Base Model**: LLaMA-2-chat 7B
-- **Training**: Fine-tuned on TOFU dataset
 
 ## Loading the Model
 
@@ -22,17 +43,14 @@ model = AutoModelForCausalLM.from_pretrained("OPTML-Group/TOFU-origin-Llama-2-7b
 
 If you use this model in your research, please cite: 
 ```
-@misc{fan2024simplicityprevailsrethinkingnegative,
-      title={Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning}, 
-      author={Chongyu Fan and Jiancheng Liu and Licong Lin and Jinghan Jia and Ruiqi Zhang and Song Mei and Sijia Liu},
-      year={2024},
-      eprint={2410.07163},
-      archivePrefix={arXiv},
-      primaryClass={cs.CL},
-      url={https://arxiv.org/abs/2410.07163}, 
+@article{fan2024simplicity,
+  title={Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning},
+  author={Fan, Chongyu and Liu, Jiancheng and Lin, Licong and Jia, Jinghan and Zhang, Ruiqi and Mei, Song and Liu, Sijia},
+  journal={arXiv preprint arXiv:2410.07163},
+  year={2024}
 }
 ```
 
-## Contact
+## Reporting Issues
 
-For questions or issues regarding this model, please contact chongyu.fan93@gmail.com.
+Reporting issues with the model: [github.com/OPTML-Group/Unlearn-Simple](https://github.com/OPTML-Group/Unlearn-Simple)