Views
No views yet
| Epoch | Validation Loss |
|---|---|
| 1 | 1.8638 |
| 2 | 1.5106 |
| 3 | 1.6593 |
| Metric | Score |
|---|---|
| ROUGE-1 | 0.4693 |
| ROUGE-2 | 0.2516 |
| ROUGE-L | 0.3786 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5base_model = AutoModelForCausalLM.from_pretrained(
6 "mistralai/Mistral-7B-v0.1",
7 load_in_4bit=True,
8 device_map="auto"
9)
10model = PeftModel.from_pretrained(base_model, "theelvace/mistral-african-fintech")
11tokenizer = AutoTokenizer.from_pretrained("theelvace/mistral-african-fintech")
12
13prompt = "### Instruction:\nWhat is the role of stablecoins in African payments?\n\n### Response:\n"
14inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
15outputs = model.generate(**inputs, max_new_tokens=200, temperature=0.7, do_sample=True)
16print(tokenizer.decode(outputs[0], skip_special_tokens=True))