Skip to main content

DeepSparse

本页介绍如何在 LangChain 中使用 DeepSparse 推理运行时。 内容分为两个部分:安装和设置,以及 DeepSparse 使用示例。

安装与设置

LLMs

存在一个 DeepSparse LLM 包装器,您可以通过以下方式访问:

from langchain_community.llms import DeepSparse

它为所有模型提供了统一的接口:

llm = DeepSparse(model='zoo:nlg/text_generation/codegen_mono-350m/pytorch/huggingface/bigpython_bigquery_thepile/base-none')

print(llm.invoke('def fib():'))

可以使用 config 参数传递额外的参数:

config = {'max_generated_tokens': 256}

llm = DeepSparse(model='zoo:nlg/text_generation/codegen_mono-350m/pytorch/huggingface/bigpython_bigquery_thepile/base-none', config=config)

此页面是否有帮助?


您还可以留下详细的反馈 在 GitHub 上