update hf

README.md

@@ -48,7 +48,7 @@ pip install -U textgen
 
 ```python
 from textgen import LlamaModel
-model = LlamaModel("llama", "shibing624/chinese-alpaca-plus-7b")
+model = LlamaModel("llama", "shibing624/chinese-alpaca-plus-7b-hf")
 r = model.predict(["用一句话描述地球为什么是独一无二的。"])
 print(r) # ['地球是独一无二的，因为它拥有独特的大气层、水循环、生物多样性以及其他自然资源，这些都使它成为一个独特的生命支持系统。']
 ```
@@ -78,8 +78,8 @@ def generate_prompt(text):
 ### Response:"""
 
 
-tokenizer = LlamaTokenizer.from_pretrained('shibing624/chinese-alpaca-plus-7b')
-model = LlamaForCausalLM.from_pretrained('shibing624/chinese-alpaca-plus-7b').half().cuda()
+tokenizer = LlamaTokenizer.from_pretrained('shibing624/chinese-alpaca-plus-7b-hf')
+model = LlamaForCausalLM.from_pretrained('shibing624/chinese-alpaca-plus-7b-hf').half().cuda()
 model.eval()
 
 text = '为什么天空是蓝色的？'
@@ -116,7 +116,7 @@ release合并后的模型权重，一次到位直接使用，省电、减少碳
 
 模型文件组成：
 ```
-chinese-alpaca-plus-7b
+chinese-alpaca-plus-7b-hf
     config.json
     generation_config.json
     pytorch_model-00001-of-00002.bin
@@ -155,4 +155,4 @@ chinese-alpaca-plus-7b
 
 
 ## Reference
-- https://github.com/ymcui/Chinese-LLaMA-Alpaca
+- https://github.com/ymcui/Chinese-LLaMA-Alpaca