.

README.md

@@ -6,7 +6,7 @@ license: apache-2.0
 
 Official research release for the family of **XGen** models (`7B`) by Salesforce AI Research:
 
-*Title*: [Long Sequence Modeling with XGen: A 7B LLM Trained on 8K Input Sequence Length](https://blog.salesforceairesearch.com/xgen/)
+*Title*: [Long Sequence Modeling with XGen: A 7B LLM Trained on 8K Input Sequence Length](https://blog.salesforceairesearch.com/xgen-7b/)
 
 ## Models
 
@@ -16,7 +16,7 @@ Official research release for the family of **XGen** models (`7B`) by Salesforce
   * License: Apache-2.0
 
 The training data for the models are tokenized with OpenAI Tiktoken library.
-To use this model, install tiktoken library via `pip`:
+To use this model, install the package via `pip`:
 
 ```sh
 pip install tiktoken
@@ -25,6 +25,7 @@ pip install tiktoken
 The models can be used as auto-regressive samplers as follows:
 
 ```python
+import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("Salesforce/xgen-7b-8k-base", trust_remote_code=True)