yarn-llama2

Llama 2 的擴展版本，支援高達 128k tokens 的上下文。

7b 13b

77.7K 15 months ago

67 標籤

Readme

Yarn Llama 2 是一個基於 Llama2 的模型，它將上下文大小擴展到 128k 上下文。它由 Nous Research 開發，通過實施 YaRN 方法來進一步訓練模型以支援更大的上下文窗口。

CLI

64k context size

ollama run yarn-llama2

128k context size

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：大型語言模型的高效上下文窗口擴展