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 |
+
'''
|