shibing624 commited on
Commit
56c8325
1 Parent(s): 34c3b73

Update README.md

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