Sponsored by
Vmake Labs Video Enhancer
THUDM / LongAlign-6B-64k
Model's Last Updated: March 07 2024
text-generation
Introduction of LongAlign-6B-64k
Model Details of LongAlign-6B-64k
LongAlign-6B-64k
🤗 [LongAlign Dataset] • 💻 [Github Repo] • 📃 [LongAlign Paper]
LongAlign is the first full recipe for LLM alignment on long context. We propose the LongAlign-10k dataset, containing 10,000 long instruction data of 8k-64k in length. We investigate on trianing strategies, namely packing (with loss weighting) and sorted batching , which are all implemented in our code. For real-world long context evaluation, we introduce LongBench-Chat that evaluate the instruction-following capability on queries of 10k-100k length.
All Models
We open-sourced the following list of models:
| Model | Huggingface Repo | Description |
|---|---|---|
| LongAlign-6B-64k-base | 🤗 Huggingface Repo | ChatGLM3-6B with an extended 64k context window |
| LongAlign-6B-64k | 🤗 Huggingface Repo | Chat model by LongAlign training on LongAlign-6B-64k-base |
| LongAlign-7B-64k-base | 🤗 Huggingface Repo | Llama-2-7B with an extended 64k context window |
| LongAlign-7B-64k | 🤗 Huggingface Repo | Chat model by LongAlign training on LongAlign-7B-64k-base |
| LongAlign-13B-64k-base | 🤗 Huggingface Repo | Llama-2-13B with an extended 64k context window |
| LongAlign-13B-64k | 🤗 Huggingface Repo | Chat model by LongAlign training on LongAlign-13B-64k-base |
| ChatGLM3-6B-128k | 🤗 Huggingface Repo | ChatGLM3-6B with a 128k context window |
Model usage
Chat prompt template for LongAlign-6B-64k:
[Round 1]
问：Hi!
答：Hello! What can I assist you today?
[Round 2]
问：What should I do if I can't sleep at night?
答：
Chat prompt template for LongAlign-7B-64k and LongAlign-13B-64k:
[INST]Hi![/INST]Hello! What can I assist you today?
[INST]What should I do if I can't sleep at night?[/INST]
ChatGLM3-6B-128k uses the same prompt template as ChatGLM3-6B .
A simple demo for deployment of the model:
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch
tokenizer = AutoTokenizer.from_pretrained("THUDM/LongAlign-6B-64k", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("THUDM/LongAlign-6B-64k", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
model = model.eval()
query = open("assets/paper.txt").read() + "\n\nPlease summarize the paper."
response, history = model.chat(tokenizer, query, history=[], max_new_tokens=512, temperature=1)
print(response)
Citation
If you find our work useful, please consider citing LongAlign:
Runs of THUDM LongAlign-6B-64k on huggingface.co
6
Total runs
0
24-hour runs
0
3-day runs
0
7-day runs
0
30-day runs
More Information About LongAlign-6B-64k huggingface.co Model
Url of LongAlign-6B-64k
LongAlign-6B-64k huggingface.co Url
Provider of LongAlign-6B-64k huggingface.co
Other API from THUDM
Total runs:
148.1K
Run Growth:
0
Growth Rate:
0.00%
Total runs:
7.7K
Run Growth:
0
Growth Rate:
0.00%
Total runs:
1.1K
Run Growth:
0
Growth Rate:
0.00%
Total runs:
1.1K
Run Growth:
0
Growth Rate:
0.00%
Total runs:
1.0K
Run Growth:
0
Growth Rate:
0.00%
Total runs:
974
Run Growth:
0
Growth Rate:
0.00%
Total runs:
924
Run Growth:
0
Growth Rate:
0.00%
Total runs:
473
Run Growth:
0
Growth Rate:
0.00%
Total runs:
353
Run Growth:
0
Growth Rate:
0.00%
Total runs:
256
Run Growth:
0
Growth Rate:
0.00%
Total runs:
218
Run Growth:
0
Growth Rate:
0.00%
Total runs:
195
Run Growth:
0
Growth Rate:
0.00%
Total runs:
76
Run Growth:
0
Growth Rate:
0.00%
Total runs:
69
Run Growth:
0
Growth Rate:
0.00%
Total runs:
49
Run Growth:
0
Growth Rate:
0.00%
Total runs:
31
Run Growth:
0
Growth Rate:
0.00%
Total runs:
20
Run Growth:
0
Growth Rate:
0.00%
Total runs:
6
Run Growth:
0
Growth Rate:
0.00%
Total runs:
6
Run Growth:
0
Growth Rate:
0.00%
Total runs:
4
Run Growth:
0
Growth Rate:
0.00%
Total runs:
4
Run Growth:
0
Growth Rate:
0.00%