Views
No views yet
google/gemma-2b model to answer legal questions, reference specific articles, and assist with compliance-related inquiries.bnb_4bit_compute_dtype=torch.bfloat16)SFTTrainer1from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
2from peft import PeftModel
3import torch
4
5# 4-bit quantization config
6bnb_config = BitsAndBytesConfig(
7 load_in_4bit=True,
8 bnb_4bit_quant_type="nf4",
9 bnb_4bit_compute_dtype=torch.bfloat16
10)
11
12# Load base model
13base_model = AutoModelForCausalLM.from_pretrained(
14 "google/gemma-2b",
15 quantization_config=bnb_config,
16 device_map="auto"
17)
18
19# Load LoRA adapter
20model = PeftModel.from_pretrained(base_model, "path_to_outputs_compliance")
21tokenizer = AutoTokenizer.from_pretrained("path_to_outputs_compliance")
22
23# Generate predictions
24input_text = "Quelle est la réglementation pour les prêts bancaires aux entreprises en Tunisie ?"
25inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
26outputs = model.generate(**inputs, max_length=200)
27print(tokenizer.decode(outputs[0], skip_special_tokens=True))