Update README.md

README.md

@@ -15,7 +15,7 @@ metrics:
 - comet
 pipeline_tag: translation
 ---
-# Model Card for TowerMistral-Instruct-v0.2
+# Model Card for TowerInstruct-Mistral-7B-v0.2
 
 ## Model Details
 
@@ -24,7 +24,7 @@ pipeline_tag: translation
 TowerInstruct-Mistral-7B-v0.2 is a language model that results from fine-tuning a Mistral version of TowerBase on the TowerBlocks supervised fine-tuning dataset.
 The model is trained to handle several translation-related tasks, such as general machine translation (e.g., sentence- and paragraph/document-level translation, terminology-aware translation, context-aware translation), automatic post edition, named-entity recognition, gramatical error correction, and paraphrase generation. 
 
-This model has performance comparable to [TowerInstruct-13B-v0.2](https://huggingface.co/Unbabel/TowerInstruct-13B-v0.1), while being half the size. Check out our [paper](https://arxiv.org/abs/2402.17733).
+This model has performance comparable to [TowerInstruct-13B-v0.2](https://huggingface.co/Unbabel/TowerInstruct-13B-v0.1), while being half the size. Check out our [paper in COLM 2024](https://openreview.net/pdf?id=EHPns3hVkj).
 
 - **Developed by:** Unbabel, Instituto Superior Técnico, CentraleSupélec University of Paris-Saclay 
 - **Model type:** A 7B parameter model fine-tuned on a mix of publicly available, synthetic datasets on translation-related tasks, as well as conversational datasets and code instructions.
@@ -58,7 +58,7 @@ Here's how you can run the model using the `pipeline()` function from 🤗 Trans
 import torch
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="Unbabel/TowerMistralInstruct-v0.2", torch_dtype=torch.bfloat16, device_map="auto")
+pipe = pipeline("text-generation", model="Unbabel/TowerInstruct-Mistral-7B-v0.2", torch_dtype=torch.bfloat16, device_map="auto")
 # We use the tokenizer’s chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
 messages = [
     {"role": "user", "content": "Translate the following text from Portuguese into English.\nPortuguese: Um grupo de investigadores lançou um novo modelo para tarefas relacionadas com tradução.\nEnglish:"},
@@ -81,11 +81,11 @@ We are currently working on improving quality and consistency on document-level
 
 ## Bias, Risks, and Limitations
 
-TowerMistral-Instruct-v0.2 has not been aligned to human preferences, so the model may generate problematic outputs (e.g., hallucinations, harmful content, or false statements).  
+TowerInstruct-Mistral-7B-v0.2 has not been aligned to human preferences, so the model may generate problematic outputs (e.g., hallucinations, harmful content, or false statements).  
 
 ## Prompt Format
 
-TowerMistral-Instruct-v0.2 was trained using the ChatML prompt templates without any system prompts. An example follows below:
+TowerInstruct-Mistral-7B-v0.2 was trained using the ChatML prompt templates without any system prompts. An example follows below:
 ```
 <|im_start|>user
 {USER PROMPT}<|im_end|>
@@ -108,13 +108,13 @@ Link to [TowerBlocks](https://huggingface.co/datasets/Unbabel/TowerBlocks-v0.1).
 ## Citation 
 
 ```bibtex
-@misc{tower_llm_2024,
-      title={Tower: An Open Multilingual Large Language Model for Translation-Related Tasks}, 
-      author={Duarte M. Alves and José Pombal and Nuno M. Guerreiro and Pedro H. Martins and João Alves and Amin Farajian and Ben Peters and Ricardo Rei and Patrick Fernandes and Sweta Agrawal and Pierre Colombo and José G. C. de Souza and André F. T. Martins},
-      year={2024},
-      eprint={2402.17733},
-      archivePrefix={arXiv},
-      primaryClass={cs.CL}
+@inproceedings{
+alves2024tower,
+title={Tower: An Open Multilingual Large Language Model for Translation-Related Tasks},
+author={Duarte Miguel Alves and Jos{\'e} Pombal and Nuno M Guerreiro and Pedro Henrique Martins and Jo{\~a}o Alves and Amin Farajian and Ben Peters and Ricardo Rei and Patrick Fernandes and Sweta Agrawal and Pierre Colombo and Jos{\'e} G. C. de Souza and Andre Martins},
+booktitle={First Conference on Language Modeling},
+year={2024},
+url={https://openreview.net/forum?id=EHPns3hVkj}
 }
 ```