Spaces:
Runtime error
Runtime error
lizhen
commited on
Commit
·
abb3004
1
Parent(s):
ea14efe
model入门,openAI的包装器。
Browse files- 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 |
+
'''
|