文心大模型文档

模型调用

用户设置

  • 在代码最开始处设置您在官网获取到的ak和sk
import wenxin_api
wenxin_api.ak = "your ak"
wenxin_api.sk = "your sk"

使用精调模型

  • 模型训练成功后,您可以调用已训练好的模型。
  • 首次调用已训练好模型,可能需要一段时间才能准备好处理请求。如果您的请求超时,很可能是因为您的模型仍在加载部署中。如果发生这种情况,请在几分钟后重试。
  • 和直接调用方式相同,用户可以复用python sdk直接调用的代码,只需将模型替换为用户自己训练好的模型
from wenxin_api import Task

model_id = "your model id"
model = Model.retrieve(model_id=model_id)
text = "标题:乌龟与野鸭。大纲:空中掉下来|头脑简单|开口说话|打算讲|飞行环游世界|乌龟厌倦|乌龟摔死|野鸭听。正文:"
result = Task.create(text=text, model=model, seq_len=512)
print(result)
  • 结果返回示例
{'result': '从前,有一只头脑简单的乌龟厌倦了自己的家,想要出去看看外面的世界。乌龟打算讲一个关于自己的飞行环游世界的故事。乌龟从自己的家里出发,一直飞到了一个湖边,在湖边,乌龟遇到了一只野鸭。野鸭听了乌龟的故事,非常羡慕乌龟,也想要去环游世界。于是,野鸭和乌龟约定,一起去环游世界。乌龟和野鸭一起乘着热气球飞行环游世界。乌龟和野鸭在空中掉下来,乌龟从空中掉下来,砸死了。'}
上一篇
模型训练
下一篇
任务示例