Update README.md (#3)

- Update README.md (1096b2aab0f9833a995211b1531b61eec6009ae3)


Co-authored-by: Ashmal Vayani <[email protected]>

README.md

@@ -39,7 +39,7 @@ prompt = "What are the psychological effects of urban living on mental health?"
 
 input_str = template.format(prompt=prompt)
 input_ids = tokenizer(input_str, return_tensors="pt").input_ids
-outputs = model.generate(input_ids, max_length=1000)
+outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
 print(tokenizer.batch_decode(outputs[:, input_ids.shape[1]:-1])[0].strip())
 ```
 
@@ -87,4 +87,6 @@ python3 -m fastchat.serve.cli --model-path MBZUAI/MobiLlama-05B-Chat
 
 
 ## Intended Uses
-Given the nature of the training data, the MobiLlama-05B model is best suited for prompts using the QA format, the chat format, and the code format.
+Given the nature of the training data, the MobiLlama-05B model is best suited for prompts using the QA format, the chat format, and the code format.
+
+## Citation