lizhen commited on
Commit
abb3004
·
1 Parent(s): ea14efe

model入门,openAI的包装器。

Browse files
Files changed (1) hide show
  1. llms_openai.py +14 -0
llms_openai.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain.llms import OpenAI
2
+
3
+ llm = OpenAI(model_name="text-davinci-003", n=2, best_of=2)
4
+
5
+ # 生成文本, get_num_tokens用于计算输入文本的token。
6
+ result = llm("随机一份双色球号码")
7
+ print()
8
+ print("随机一份双色球号码. Number of Tokens:%s"%(llm.get_num_tokens("随机一份双色球号码")), result)
9
+
10
+ '''
11
+ # generate,输入是一个list, 返回多条详细回复list
12
+ list = llm.generate(["随机一份双色球号码","取出红球号码"])
13
+ print("双色球generate:", list)
14
+ '''