Update README.md

README.md

@@ -19,6 +19,23 @@ The **Yi** series models are large language models trained from scratch by devel
 
 - 🎯 **2023/11/02**: The base model of `Yi-6B` and `Yi-34B` 
 
+## Dependency Installation
+
+```shell
+pip install transformers==4.34.0 sentencepiece==0.1.99 accelerate==0.24.1
+```
+
+## Generation Demonstration
+
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-6B", device_map="auto", torch_dtype="auto", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-6B", trust_remote_code=True)
+inputs = tokenizer('Please count number for me: 1, 2, 3', return_tensors="pt")
+outputs = model.generate(inputs.input_ids.cuda(), max_new_tokens=256)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
 ## Model Performance
 
 | Model         |   MMLU   |  CMMLU   |  C-Eval  |  GAOKAO  |   BBH    | Commonsense Reasoning | Reading Comprehension | Math & Code |