shibing624 commited on
Commit
6c1958d
1 Parent(s): 0d2834a

Update README.md

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