shibing624 commited on
Commit
bc01daa
1 Parent(s): f62f120

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -56,8 +56,8 @@ pip install -U textgen
56
  ```
57
 
58
  ```python
59
- from textgen import LlamaModel
60
- model = LlamaModel("llama", "shibing624/chinese-llama-plus-13b-hf")
61
  r = model.predict(["用一句话描述地球为什么是独一无二的。"])
62
  print(r) # ['地球是独一无二的,因为它拥有独特的大气层、水循环、生物多样性以及其他自然资源,这些都使它成为一个独特的生命支持系统。']
63
  ```
 
56
  ```
57
 
58
  ```python
59
+ from textgen import GptModel
60
+ model = GptModel("llama", "shibing624/chinese-llama-plus-13b-hf")
61
  r = model.predict(["用一句话描述地球为什么是独一无二的。"])
62
  print(r) # ['地球是独一无二的,因为它拥有独特的大气层、水循环、生物多样性以及其他自然资源,这些都使它成为一个独特的生命支持系统。']
63
  ```