文档
基础知识
推测解码
在 lmstudio-js
中使用推测解码草稿模型的 API
推测解码是一种可以显著提高大型语言模型 (LLM) 生成速度而又不降低响应质量的技术。有关更多信息,请参阅推测解码。
要在 lmstudio-js
中使用推测解码,只需在执行预测时提供 draftModel
参数即可。您无需单独加载草稿模型。
import { LMStudioClient } from "@lmstudio/sdk";
const client = new LMStudioClient();
const mainModelKey = "qwen2.5-7b-instruct";
const draftModelKey = "qwen2.5-0.5b-instruct";
const model = await client.llm.model(mainModelKey);
const result = await model.respond("What are the prime numbers between 0 and 100?", {
draftModel: draftModelKey,
});
const { content, stats } = result;
console.info(content);
console.info(`Accepted ${stats.acceptedDraftTokensCount}/${stats.predictedTokensCount} tokens`);
此页面的源代码可在 GitHub 上获取