Views
No views yet
<s> [INST] Instruction [/INST] Model answer</s> [INST] Follow-up instruction [/INST]<s> and </s> are special tokens for beginning of string (BOS) and end of string (EOS) while [INST] and [/INST] are regular strings.1def tokenize(text):
2 return tok.encode(text, add_special_tokens=False)
3
4[BOS_ID] +
5tokenize("[INST]") + tokenize(USER_MESSAGE_1) + tokenize("[/INST]") +
6tokenize(BOT_MESSAGE_1) + [EOS_ID] +
7…
8tokenize("[INST]") + tokenize(USER_MESSAGE_N) + tokenize("[/INST]") +
9tokenize(BOT_MESSAGE_N) + [EOS_ID]tokenize method should not add a BOS or EOS token automatically, but should add a prefix space.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5
6model = AutoModelForCausalLM.from_pretrained(model_id)
7
8text = "Hello my name is"
9inputs = tokenizer(text, return_tensors="pt")
10
11outputs = model.generate(**inputs, max_new_tokens=20)
12print(tokenizer.decode(outputs[0], skip_special_tokens=True))float16 precision only works on GPU devices1+ import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6
7+ model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16).to(0)
8
9text = "Hello my name is"
10+ inputs = tokenizer(text, return_tensors="pt").to(0)
11
12outputs = model.generate(**inputs, max_new_tokens=20)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))bitsandbytes1+ import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6
7+ model = AutoModelForCausalLM.from_pretrained(model_id, load_in_4bit=True)
8
9text = "Hello my name is"
10+ inputs = tokenizer(text, return_tensors="pt").to(0)
11
12outputs = model.generate(**inputs, max_new_tokens=20)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))1+ import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6
7+ model = AutoModelForCausalLM.from_pretrained(model_id, use_flash_attention_2=True)
8
9text = "Hello my name is"
10+ inputs = tokenizer(text, return_tensors="pt").to(0)
11
12outputs = model.generate(**inputs, max_new_tokens=20)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))