--- base_model: ismaprasetiyadi/Biawak-8B-Base library_name: peft pipeline_tag: text-generation language: - id tags: - base_model:adapter:ismaprasetiyadi/Biawak-8B-Base - lora - sft - transformers - trl - unsloth - biawak - indonesian - instruction-following license: apache-2.0 datasets: - dtp-fine-tuning/dtp-singleturn-AGQ-9k --- # Model Card for SFT-Biawak-8B-AGQ-9k-Unsloth ## Model Details ### Model Description This model is a fine-tuned version of **[ismaprasetiyadi/Biawak-8B-Base](https://huggingface.co/ismaprasetiyadi/Biawak-8B-Base)**. It was trained using **Unsloth** and **LoRA** (Low-Rank Adaptation) on the **[dtp-singleturn-AGQ-9k](https://huggingface.co/datasets/dtp-fine-tuning/dtp-singleturn-AGQ-9k)** dataset. The model is specifically optimized for **Indonesian single-turn instruction following**, utilizing the Qwen3 chat template structure. It leverages 4-bit quantization for memory efficiency during training and inference. - **Developed by:** DTP2 Team - **Model type:** Causal Language Model (LoRA Adapter) - **Language(s) (NLP):** Indonesian (id) - **License:** Apache-2.0 - **Finetuned from model:** [ismaprasetiyadi/Biawak-8B-Base](https://huggingface.co/ismaprasetiyadi/Biawak-8B-Base) ### Model Sources - **Repository:** [dtp-fine-tuning/DTP_AGQ_Question_Diploy_9K](https://huggingface.co/dtp-fine-tuning/DTP_AGQ_Question_Diploy_9K) - **Dataset:** [dtp-fine-tuning/dtp-singleturn-AGQ-9k](https://huggingface.co/datasets/dtp-fine-tuning/dtp-singleturn-AGQ-9k) - **Training Logs:** [View full W&B Report](https://api.wandb.ai/links/DTP2/zh541s0e) ## Uses ### Direct Use The model is designed for Indonesian chat and instruction-following tasks. It performs best in single-turn question-answering scenarios involving general knowledge, reasoning, and cultural context provided by the AGQ dataset. ### Out-of-Scope Use - **Long-context conversations:** The model was fine-tuned on single-turn data; multi-turn performance may be inconsistent. - **High-stakes decision making:** As an 8B model, it may hallucinate facts and should not be used for medical or legal advice without verification. ## Bias, Risks, and Limitations This model inherits the biases present in the base `Biawak-8B` model and the `AGQ-9k` dataset. While fine-tuning improves instruction adherence, users should be aware that the model can still generate plausible-sounding but incorrect information. ### Recommendations Users should verify important information generated by the model. It is recommended to use the `qwen3` chat template for optimal performance. ## How to Get Started with the Model Use the code below to load the model and run inference: ```python from transformers import AutoModelForCausalLM, AutoTokenizer from peft import PeftModel import torch # 1. Load Base Model base_model_name = "ismaprasetiyadi/Biawak-8B-Base" adapter_model_name = "YOUR_USERNAME/SFT-Biawak-8B-AGQ-9k-Unsloth" model = AutoModelForCausalLM.from_pretrained( base_model_name, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True ) tokenizer = AutoTokenizer.from_pretrained(adapter_model_name, trust_remote_code=True) # 2. Load Adapter model = PeftModel.from_pretrained(model, adapter_model_name) # 3. Inference messages = [ {"role": "user", "content": "Jelaskan sejarah singkat kemerdekaan Indonesia."} ] text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True) inputs = tokenizer(text, return_tensors="pt").to("cuda") outputs = model.generate(**inputs, max_new_tokens=512) print(tokenizer.decode(outputs[0], skip_special_tokens=True)) ``` ## Training Details ### Training Data The model was trained on dtp-fine-tuning/dtp-singleturn-AGQ-9k. - Size: ~9k examples - Content: Indonesian general questions and instructions (Single Turn). - Split: Train (90%) / Test (10%) ### Training Procedure The model was fine-tuned using the Unsloth library, which provides 2x faster training and ~60% less memory usage compared to standard Hugging Face implementations. #### Training Hyperparameters - **Training regime**: bf16 mixed precision (via Unsloth/LoRA) - **Quantization**: 4-bit (nf4) - **LoRA Rank**: 16 - **LoRA Alpha**: 32 - **Target Modules**: q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj - **Batch Size**: 8 per device (Effective batch size: 32 via Gradient Accumulation) - **Learning Rate**: 2e-5 (Linear Schedule with 0.05 warmup) - **Epochs**: 2 - **Max Sequence Length**: 8192 - **Optimizer**: adamw_8bit #### Speeds, Sizes, Times [optional] - **Hardware**: 1x NVIDIA A100 80GB - **Training Duration**: ~10 hours - **GPU Memory Usage**: Peaked at ~45GB (56% utilization) ## Evaluation ### Results The model demonstrated stable convergence over 2 epochs. * **Final Training Loss:** ~0.70 * **Final Validation Loss:** ~0.67 * **Observation:** The validation loss consistently decreased alongside training loss, indicating no overfitting occurred during the training process. **[View the full training run plots and metrics on Weights & Biases](https://api.wandb.ai/links/DTP2/zh541s0e)** ## Environmental Impact Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700). - **Hardware Type**: NVIDIA A100 80GB - **Hours used**: 10 hours - **Cloud Provider**: University Server / Private Infrastructure - **Compute Region**: Indonesia ### Framework versions - Unsloth 2024.x - Transformers 4.x - Pytorch 2.x - Datasets 2.x - Tokenizers 0.x