Views
No views yet
| Base model | unsloth/gemma-3-1b-it |
| Dataset | csebuetnlp/xlsum |
| Fine-tuning | Full SFT (no LoRA) |
| Framework | Unsloth + TRL |
| Epochs | ~1.48 (checkpoint-4000, best val ROUGE-L) |
| Max seq length | 3072 |
| Batch size | 8 per device |
| Learning rate | 2e-5 (cosine decay, warmup 3 %) |
| Precision | bfloat16 |
| Optimizer | adamw_8bit |
| Best eval ROUGE-L | 22.23 |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_id = "nuinashco/gemma-3-1b-it-xlsum-ua-sft"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id, torch_dtype=torch.bfloat16, device_map="auto"
8)
9
10article = "Ваш текст новини тут..."
11prompt = [
12 {"role": "user", "content": f"Зроби короткий переказ наступного тексту:\n{article}"}
13]
14inputs = tokenizer.apply_chat_template(
15 prompt, add_generation_prompt=True, return_tensors="pt"
16).to(model.device)
17
18out = model.generate(inputs, max_new_tokens=128, do_sample=False)
19print(tokenizer.decode(out[0][inputs.shape[-1]:], skip_special_tokens=True))