Mistral 的延伸版本,支援 64K 或 128K 的上下文視窗。

7b

45.2K 16 個月前

說明文件

Yarn Mistral 是一個基於 Mistral 的模型,其擴展了上下文大小至 128k 上下文。它由 Nous Research 開發,透過實作 YaRN 方法,進一步訓練模型以支援更大的上下文視窗。

CLI (命令行介面)

64k 上下文大小

ollama run yarn-mistral

128k 上下文大小

ollama run yarn-mistral:7b-128k

API (應用程式介面)

範例

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

參考文獻

Hugging Face

YaRN:大型語言模型的高效上下文視窗擴展