Update README.md

README.md

@@ -106,7 +106,7 @@ TeleChat模型相比同规模模型在评测效果方面也有较好的表现，
 ```python
 import os
 import torch
-from modelscope import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
+from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 os.environ["CUDA_VISIBLE_DEVICES"] = '0'
 tokenizer = AutoTokenizer.from_pretrained('TeleAI/TeleChat-12B')
 model = AutoModelForCausalLM.from_pretrained('TeleAI/TeleChat-12B', trust_remote_code=True, device_map="auto", torch_dtype=torch.float16)