An Indonesian conversational AI model fine-tuned from GPT2-Small (124M Parameters) using instruction-following techniques to enable natural chat-like interactions in Bahasa Indonesia.
📋 Model Overview
This model transforms a base Indonesian GPT-2 text generator into a conversational chatbot capable of following instructions and engaging in question-answering dialogues. The model has been specifically optimized for Indonesian language understanding and generation.
Base Model: GPT2-Small (124M Parameters)
Fine-tuning Method: SFT-LoRA (Supervised Fine-Tuning with Low-Rank Adaptation)
This model was developed as part of a personal learning journey in AI and Large Language Models (LLMs). The entire training process was conducted on Google Colab's free tier using T4 GPU, demonstrating how to build effective Indonesian conversational AI with limited computational resources.
The project focuses on creating an accessible Indonesian language model that can understand context, follow instructions, and provide helpful responses in natural Bahasa Indonesia.
Mixed Precision: FP16 enabled - Memory and speed optimization
Training Progress
The model showed consistent improvement throughout training:
Training Progress (5535 total steps over 3 epochs):
Step Training Loss
200 3.533500 # Initial high loss
400 2.964200 # Rapid initial improvement
...
4000 2.416200 # Stable convergence
...
5400 2.397500 # Final optimized loss
Final Metrics:
- Training Loss: 2.573
- Training Time: 3.5 hours
- Samples per Second: 14.049
- Total Training Samples: ~177k
The steady decrease from 3.53 to 2.39 demonstrates effective learning and adaptation to the Indonesian instruction-following task.
🔧 Advanced Usage
Generation Parameter Tuning
For Creative Responses:
python
1outputs = model.generate(2**inputs,3 max_new_tokens=256,# Longer responses4 temperature=0.8,# More randomness5 top_p=0.9,# Diverse vocabulary6 repetition_penalty=1.2# Avoid repetition7)
This model is released under the MIT License, Please see the LICENSE file for complete terms.
🙏 Acknowledgments
Base Model: Thanks to Cahya for the Indonesian GPT-2 base model
Datasets: FreedomIntelligence team for Indonesian instruction and conversation datasets
Infrastructure: Google Colab for providing accessible GPU resources for training
Disclaimer: This model was developed as an experimental project for educational and research purposes. While it demonstrates good performance on various tasks, users should validate outputs for critical applications and be aware of the limitations outlined above.