Update README.md

README.md

@@ -1,3 +1,101 @@
 ---
-license: unknown
+language:
+- zh
+- en
+tags:
+- MachineMindset
+- MBTI
+pipeline_tag: text-generation
+inference: false
+
+
 ---
+
+
+
+<p align="center">
+    <img src="https://raw.githubusercontent.com/PKU-YuanGroup/Machine-Mindset/main/images/logo.png" width="650" style="margin-bottom: 0.2;"/>
+<p>
+<h2 align="center"> <a href="https://arxiv.org/abs/2311.10122">Machine Mindset: An MBTI Exploration of Large Language Models</a></h2>
+<h5 align="center"> If you like our project, please give us a star ⭐  </h2>
+
+<br>
+
+### Introduction
+
+**MM_en_ENFP (Machine_Mindset_en_ENFP)** is an English large language model developed through a collaboration between FarReel AI Lab and Peking University Deep Research Institute, based on Llama2-7b-chat-hf with an MBTI personality type of ENFP. 
+
+MM_en_ENFP has undergone extensive training, including the creation of a large-scale MBTI dataset, multi-stage fine-tuning, and DPO training. We are committed to continuously updating the model to improve its performance and regularly supplementing it with test data. This repository serves as the storage for the MM_en_ENFP model.
+
+The foundational personality trait of **MM_en_ENFP (Machine_Mindset_en_ENFP)** is **ENFP**. This means that it tends to exhibit traits of extraversion, intuition, feeling, and perception, and detailed characteristics can be found in [16personalities](https://www.16personalities.com/).
+
+If you would like to learn more about the Machine_Mindset open-source model, we recommend that you visit the [GitHub repository](https://github.com/PKU-YuanGroup/Machine-Mindset/) for additional details.<br>
+
+### Requirements
+
+* python 3.8 and above
+* pytorch 1.12 and above, 2.0 and above are recommended
+* CUDA 11.4 and above are recommended (this is for GPU users, flash-attention users, etc.)
+  <br>
+
+### Dependency
+
+
+<br>
+
+### Quickstart
+
+* Use HuggingFace Transformers:
+  ```python
+  import torch
+  from transformers import AutoModelForCausalLM, AutoTokenizer
+  from transformers.generation.utils import GenerationConfig
+  tokenizer = AutoTokenizer.from_pretrained("FarReelAILab/Machine_Mindset_en_ENFP", use_fast=False, trust_remote_code=True)
+  model = AutoModelForCausalLM.from_pretrained("FarReelAILab/Machine_Mindset_en_ENFP", device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
+  model.generation_config = GenerationConfig.from_pretrained("FarReelAILab/Machine_Mindset_en_ENFP")
+  messages = []
+  messages.append({"role": "user", "content": "你最喜欢读哪一本书？"})
+  response = model.chat(tokenizer, messages)
+  print(response)
+  #我最喜欢的一本书是《人类简史》。这本书以独特的视角探索了人类历史的各个方面，包括文化、社会和科学的发展。它挑战了我对世界的认知，并激发了我对人类的潜力和未来发展的思考。
+  ```
+* Use LLaMA-Factory
+  ```bash
+  git clone https://github.com/hiyouga/LLaMA-Factory.git
+  cd LLaMA-Factory
+  python ./src/cli_demo.py \
+        --model_name_or_path /path_to_your_local_model \
+        --template baichuan2
+  ```
+
+For more information, please refer to our [GitHub repo](https://github.com/PKU-YuanGroup/Machine-Mindset/).
+<br>
+
+
+### Citation
+
+If you find our work helpful, feel free to give us a cite.
+
+```
+@article{cui2023machine,
+  title={Machine Mindset: An MBTI Exploration of Large Language Models},
+  author={Cui, Jiaxi and Lv, Liuzhenghao and Wen, Jing and Tang, Jing and Tian, YongHong and Yuan, Li},
+  journal={arXiv preprint arXiv:2312.12999},
+  year={2023}
+}
+```
+
+<br>
+
+### License Agreement
+Our code follows the Apache 2.0 open-source license. Please check [LICENSE](https://github.com/PKU-YuanGroup/Machine-Mindset/blob/main/LICENSE) for specific details regarding the open-source agreement.
+
+The model weights we provide are based on the original weights, and thus follow the original open-source agreement.
+
+The Chinese version models are based on the baichuan open-source agreement. It is suitable for commercial use. You can refer to [model_LICENSE](https://huggingface.co/JessyTsu1/Machine_Mindset_en_ENFP/resolve/main/Machine_Mindset%E5%9F%BA%E4%BA%8Ebaichuan%E7%9A%84%E6%A8%A1%E5%9E%8B%E7%A4%BE%E5%8C%BA%E8%AE%B8%E5%8F%AF%E5%8D%8F%E8%AE%AE.pdf) for specific details.
+
+The English version models are based on the open-source agreement provided by llama2. You can refer to [llama2 open-source license](https://ai.meta.com/resources/models-and-libraries/llama-downloads/).
+
+### Contact Us
+
+Feel free to send an email to [email protected], [email protected]