bababoi_test

A fine-tuned version of TinyLlama/TinyLlama-1.1B-Chat-v1.0 using LoRA (Low-Rank Adaptation) on a small instruction-following dataset.

Model Details

Property Value
Base Model TinyLlama/TinyLlama-1.1B-Chat-v1.0
Fine-tuning Method LoRA (Low-Rank Adaptation)
LoRA Rank 16
LoRA Alpha 16
Trainable Parameters ~12.6M (1.13% of total)
Dataset Size 20 instruction-following examples
Training Epochs 10
Learning Rate 2e-4
Hardware Apple M4 GPU (MPS)

Training Details

This model was trained on a tiny Alpaca-format dataset covering basic factual Q&A, grammar, translation, and simple arithmetic. It is intended as a tutorial/demo for learning fine-tuning workflows rather than production use.

Training Scripts

  • train.py โ€” Python training script with LoRA + HuggingFace Trainer
  • inference.py โ€” Interactive inference script
  • train_cli.sh / infer_cli.sh โ€” Shell wrappers

How to Use

Load the LoRA Adapter

from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer

base_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
adapter = "fkarnagi/bababoi_test"

model = AutoModelForCausalLM.from_pretrained(base_model, torch_dtype="auto")
model = PeftModel.from_pretrained(model, adapter)
tokenizer = AutoTokenizer.from_pretrained(adapter)

Inference

prompt = "What is the capital of France?"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=64)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

Limitations

  • Trained on only 20 examples โ€” will overfit and memorize answers
  • Not suitable for production or real-world tasks
  • Designed for educational purposes only

Acknowledgements

Framework versions

  • PEFT 0.19.1
Downloads last month
5
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for fkarnagi/bababoi_test

Adapter
(1589)
this model