Views
No views yet
tbilisi-ai-lab/kona2-small-3.8B that lets the model converse in the voice of Ilia Chavchavadze (1837–1907) — 19th-century Georgian writer, poet, publicist, and national leader known as "ერის მამა" (Father of the Nation).| Field | Value |
|---|---|
| Developed by | Giorgi Duchidze (@Duchidzeg) |
| Model type | LoRA adapter over a Phi-3.5-mini-instruct causal LM |
| Language | Georgian (ქართული) |
| License | Apache 2.0 |
| Base model | tbilisi-ai-lab/kona2-small-3.8B |
| Training data | Duchidzeg/ilia-persona-sharegpt |
repetition_penalty ≈ 1.15 and temperature ∈ [0.3, 0.6] to avoid the base model's tendency to loop on longer outputs.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
3from peft import PeftModel
4
5BASE = "tbilisi-ai-lab/kona2-small-3.8B"
6ADAPTER = "Duchidzeg/kona2-ilia"
7
8bnb = BitsAndBytesConfig(
9 load_in_4bit=True,
10 bnb_4bit_quant_type="nf4",
11 bnb_4bit_use_double_quant=True,
12 bnb_4bit_compute_dtype=torch.bfloat16,
13)
14
15tok = AutoTokenizer.from_pretrained(BASE, trust_remote_code=True)
16model = AutoModelForCausalLM.from_pretrained(
17 BASE,
18 quantization_config=bnb,
19 device_map="auto",
20 trust_remote_code=True,
21 attn_implementation="eager",
22)
23model = PeftModel.from_pretrained(model, ADAPTER)
24model.eval()
25
26SYSTEM = (
27 "შენ ხარ ილია ჭავჭავაძე, მეცხრამეტე საუკუნის ქართველი მწერალი, "
28 "პოეტი, პუბლიცისტი და ეროვნული მოძრაობის წინამძღოლი, რომელსაც ერმა "
29 "„ერის მამა“ უწოდა. საუბრობ დინჯი, დარბაისლური, მეცხრამეტე საუკუნის "
30 "ლიტერატურული ქართულით."
31)
32
33messages = [
34 {"role": "system", "content": SYSTEM},
35 {"role": "user", "content": "რას ფიქრობთ ქართული ენის მდგომარეობაზე?"},
36]
37inputs = tok.apply_chat_template(
38 messages,
39 add_generation_prompt=True,
40 return_tensors="pt",
41 return_dict=True,
42).to(model.device)
43
44out = model.generate(
45 **inputs,
46 max_new_tokens=256,
47 do_sample=True,
48 temperature=0.5,
49 top_p=0.9,
50 repetition_penalty=1.15,
51 use_cache=False,
52 pad_token_id=tok.pad_token_id,
53)
54print(tok.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))Duchidzeg/ilia-persona-sharegpt — 3,154 ShareGPT-format conversations covering Ilia's essays, prose, and poetry, plus identity and theme reinforcement across categories: identity, character disambiguation, meta/AI probes, time-period grounding, language, homeland, education, own-works attribution, modern-topic deflection, and style.-100).r = 16, lora_alpha = 32, lora_dropout = 0.0, use_rslora = Trueqkv_proj, o_proj, gate_up_proj, down_proj (Phi-3 all-linear)apply_chat_template)<|im_start|>assistant\n … <|im_end|> spans| Parameter | Value |
|---|---|
| Precision | bf16 mixed (compute), NF4 4-bit (storage) |
| Per-device batch size | 1 |
| Gradient accumulation | 8 (effective batch = 8) |
| Epochs | 3 |
| Learning rate | 1.5e-4, cosine schedule |
| Warmup | ~10% of steps per epoch |
| Weight decay | 0.01 |
| Optimizer | adamw_bnb_8bit |
| Gradient checkpointing | on (non-reentrant) |
| NEFTune noise α | 5 |
| Seed | 3407 |
load_best_model_at_end on eval_loss1@misc{duchidze2026kona2ilia,
2 title = {kona2-ilia: An Ilia Chavchavadze persona LoRA adapter for kona2-small-3.8B},
3 author = {Duchidze, Giorgi},
4 year = {2026},
5 howpublished = {\url{https://huggingface.co/Duchidzeg/kona2-ilia}},
6}