Update README.md

README.md

@@ -88,11 +88,9 @@ To select a language pair for translation, include one of the following tags in
 ```
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-
 tokenizer = AutoTokenizer.from_pretrained("nvidia/Riva-Translate-4B-Instruct-v1.1")
 model = AutoModelForCausalLM.from_pretrained("nvidia/Riva-Translate-4B-Instruct-v1.1").cuda()
 
-
 # Use the prompt template (along with chat template)
 messages = [
     {
@@ -101,10 +99,8 @@ messages = [
     },
     {"role": "user", "content": "The GRACE mission is a collaboration between the NASA and German Aerospace Center.?"},
  ]
-
 tokenized_chat = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(model.device)
 outputs = model.generate(tokenized_chat,  max_new_tokens=128, pad_token_id=tokenizer.eos_token_id)
-
 print(tokenizer.decode(outputs[0]))
 ```
 
@@ -188,7 +184,6 @@ curl http://localhost:8000/v1/chat/completions \
 ```
 
 ### Chat Template Structure
-
 ```
 {%- set language_pairs = {
   'en-zh-cn': {'source': 'English', 'target': 'Simplified Chinese'},