Views
No views yet
moncefem/Mistral-7B-v0.3-Legal-Competition[INST] ... [/INST] chat template<s>[INST] User question [/INST] Assistant answer</s>1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_name = "moncefem/Mistral-7B-v0.3-Legal-Competition-Instruct"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(
7 model_name,
8 device_map="auto",
9 torch_dtype="auto"
10)
11
12messages = [
13 {"role": "user", "content": "Quelle est la sanction maximale de l'Autorité de la concurrence ?"}
14]
15
16prompt = tokenizer.apply_chat_template(
17 messages,
18 tokenize=False,
19 add_generation_prompt=True
20)
21
22inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
23
24outputs = model.generate(
25 **inputs,
26 max_new_tokens=500,
27 temperature=0.7,
28 top_p=0.9
29)
30
31print(tokenizer.decode(outputs[0], skip_special_tokens=True))