Views
No views yet
meta-llama/Llama-3.1-8B-Instruct1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5base = AutoModelForCausalLM.from_pretrained(
6 'meta-llama/Llama-3.1-8B-Instruct', torch_dtype=torch.bfloat16, device_map='auto'
7)
8model = PeftModel.from_pretrained(base, 'khadimeli/auditflow-ohada-llama3-dpo')
9tok = AutoTokenizer.from_pretrained('khadimeli/auditflow-ohada-llama3-dpo')
10
11messages = [
12 {"role": "system", "content": "Tu es un expert en audit OHADA/Sénégal."},
13 {"role": "user", "content": "La loi de Benford est violée (chi²=7698). ISA 240 ??"},
14]
15ids = tok.apply_chat_template(messages, add_generation_prompt=True, return_tensors='pt')
16out = model.generate(ids, max_new_tokens=300)
17print(tok.decode(out[0][ids.shape[-1]:], skip_special_tokens=True))