Llama 2 的扩展版本,支持高达 128k 个token 的上下文。

7b 13b

72.4K 12 个月前

12 个月前

0afb4364db77 · 5.4GB

模型
llama
·
13B
·
Q2_K
参数
{"num_ctx":65536}

自述文件

Yarn Llama 2 是基于 Llama2 的模型,它将上下文大小扩展到 128k 个token。该模型由 Nous Research 开发,通过实现 YaRN 方法来进一步训练模型,使其支持更大的上下文窗口。

CLI

64k 个token 的上下文大小

ollama run yarn-llama2

128k 个token 的上下文大小

ollama run yarn-llama2:7b-128k

API

示例

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

参考资料

Hugging Face

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