该模型将LLama-3 8B的上下文长度从8k扩展到超过100万个token。

8b 70b

89.5K 6个月前

自述文件

该模型将LLama-3 8B的上下文长度从8k扩展到超过1040K,由Gradient开发,由Crusoe Energy提供计算资源赞助。它表明,通过适当地调整RoPE theta,SOTA LLM可以学习在长上下文中运行,并且只需进行最少的训练。我们在此阶段训练了8.3亿个token,在所有阶段总共训练了14亿个token,这不到LLama-3原始预训练数据的0.01%。

大型上下文窗口

注意:使用256k的上下文窗口至少需要64GB的内存。使用1M+的上下文窗口需要更多内存(100GB+)。

要扩展上下文窗口(例如扩展到256k),请使用

API

curl https://127.0.0.1:11434/api/generate -d '{
  "model": "llama3-gradient",
  "prompt": "Why is the sky blue?",
  "options": {
    "num_ctx": 256000
  }
}'

CLI

ollama run llama3-gradient
>>> /set parameter num_ctx 256000

参考文献

网站

Hugging Face