Views
No views yet
1{
2 "rope_type": "yarn",
3 "factor": 2.0,
4 "original_max_position_embeddings": 4096,
5 "beta_fast": 32,
6 "beta_slow": 1
7}1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained(
4 "DeL-TaiseiOzaki/baseline-model-instruct-8k-yarn",
5 torch_dtype="bfloat16",
6 device_map="auto"
7)
8tokenizer = AutoTokenizer.from_pretrained("DeL-TaiseiOzaki/baseline-model-instruct-8k-yarn")