Tele-AI
/

TeleChat-12B-v2

Text Generation

Model card Files Files and versions Community

liusx commited on May 20

Commit

81bcc8e

•

1 Parent(s): 1b6f3f8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -106,7 +106,7 @@ TeleChat模型相比同规模模型在评测效果方面也有较好的表现，
 ```python
 import os
 import torch
-from modelscope 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)

 ```python
 import os
 import torch
+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)