Update README.md

README.md

@@ -27,7 +27,7 @@ If you find this model helpful, please *like* this model and star us on https://
 ### 模型使用
 ``` python
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 model_dir = '/path/to_finetuned_model/'
 img_path = 'you_image_path'
 tokenizer = AutoTokenizer.from_pretrained(model_dir, trust_remote_code=True)