Update README.md

README.md

@@ -4,15 +4,15 @@ license: apache-2.0
 单论对话机器人
 调用方式
 
-from transformers import BloomForCausalLM, BloomTokenizerFast
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 checkpoint = "model"
 
 
-tokenizer = BloomTokenizerFast.from_pretrained(checkpoint)
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 
-model = BloomForCausalLM.from_pretrained(checkpoint).cuda()
+model = AutoModelForCausalLM.from_pretrained(checkpoint).cuda()
 
 text = 'Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{你想问的文本}\n\n### Response:'