yarn-mistral

Mistral 的扩展，支持 64K 或 128K 的上下文窗口。

7b

41.7K 12 个月前

33 个标签

12 个月前

24fcccd3ec98 · 14GB

模型
llama
·
7.24B
·
F16
参数
{"num_ctx":131072}

自述文件

Yarn Mistral 是基于 Mistral 的模型，将其上下文大小扩展到 128k。它是由 Nous Research 通过实现 YaRN 方法来进一步训练模型以支持更大的上下文窗口而开发的。

CLI

64k 上下文大小

ollama run yarn-mistral

128k 上下文大小

ollama run yarn-mistral:7b-128k

API

示例

curl -X POST https://127.0.0.1:11434/api/generate -d '{
  "model": "yarn-mistral:7b-128k",
  "prompt":"Here is a story about llamas eating grass"
 }'

参考

Hugging Face

YaRN：大型语言模型的有效上下文窗口扩展