Update README.md

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