PJMixers-Dev/Granite-3.1-Earthen-v0.3-3B-A800M
PJMixers-Dev/Granite-3.1-Earthen-v0.3-3B-A800M is a 3 billion parameter instruction-tuned causal language model based on IBM's Granite-3.1 architecture, developed by PJMixers-Dev. This model was fine-tuned using QLoRA with a context length of 8192 tokens, focusing on a diverse dataset including story-like data, subtitles, news, and various instruction and role-play datasets. It is designed for general conversational tasks and creative text generation, leveraging an optimized training configuration with the CAME optimizer and Liger kernels.
Loading preview...
Model Overview
PJMixers-Dev/Granite-3.1-Earthen-v0.3-3B-A800M is a 3 billion parameter language model derived from the ibm-granite/granite-3.1-3b-a800m-instruct base. This version represents a test run, fine-tuned using QLoRA with a focus on diverse text generation capabilities. It was trained for 400 steps, processing over 52 million unique tokens, and supports a context length of 8192 tokens.
Training and Optimization
The model was fine-tuned using the Axolotl framework, incorporating several advanced optimization techniques:
- QLoRA Adaptation: Utilizes QLoRA with a rank of 128 and alpha of 128 for efficient fine-tuning.
- Optimized Training: Employs the
came_pytorchoptimizer with stochastic rounding, cautious mode, and 8-bit precision enabled, alongside the REX learning rate scheduler. - Kernel Optimizations: Integrates Liger kernels for Rope, RMS Norm, Layer Norm, GLU activation, and cross-entropy, enhancing training efficiency.
- Diverse Dataset: Training data includes a wide array of sources such as story-like content (e.g., BeaverAI, PJMixers-Dev/Lit-axo-Shuffled), subtitles, news articles (AP-News), and various instruction-following and role-play datasets (e.g., allenai/tulu-3-sft-mixture, Gryphe/Aesir-RPG-Charcards).
Prompt Format
This model adheres to the Granite-3.1 Instruct format, using specific tokens for roles:
<|start_of_role|>system<|end_of_role|>example system prompt<|end_of_text|>
<|start_of_role|>user<|end_of_role|>example user turn 1<|end_of_text|>
<|start_of_role|>assistant<|end_of_role|>example assistant turn 1<|end_of_text|>Use Cases
This model is suitable for:
- General Instruction Following: Responding to user prompts in a conversational manner.
- Creative Text Generation: Generating story-like content, dialogues, and role-play scenarios.
- Text Completion: Leveraging its training on diverse completion datasets for various writing tasks.