Odia Mistral 7B DAFT Model

This model is a fine-tuned version of Mistral-7B for Odia (ଓଡ଼ିଆ) text generation using DAFT (Discriminative Alignment-Free Training).

Model Details

Base Model: unsloth/mistral-7b-bnb-4bit
Training Method: DAFT (Discriminative Alignment-Free Training)
Language: Odia (ଓଡ଼ିଆ)
Training Data: 25,000 Odia text examples
Training Steps: 3,125
Framework: Unsloth (2x faster training)

Training Details

Method: LoRA fine-tuning with r=16
Batch Size: 2 (effective: 8 with gradient accumulation)
Learning Rate: 1e-4
Optimizer: AdamW 8-bit
Precision: FP16

Usage

from unsloth import FastLanguageModel

# Load model
model, tokenizer = FastLanguageModel.from_pretrained(
    model_name="abhilash88/odia-mistral-7b-daft",
    max_seq_length=400,
    dtype=torch.float16,
    load_in_4bit=True,
)

# Enable inference mode
FastLanguageModel.for_inference(model)

# Generate Odia text
prompt = "ଓଡ଼ିଆରେ ଲେଖ:"
formatted_prompt = f"<|user|>\n{prompt}\n<|assistant|>\n"

inputs = tokenizer(formatted_prompt, return_tensors="pt").to("cuda")
outputs = model.generate(**inputs, max_new_tokens=50, temperature=0.7)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)

print(response.split("<|assistant|>")[-1].strip())

Sample Outputs

Prompt: ଲେଖ: (Write) Output: [Your model's generated Odia text]

Performance

Successfully trained on 25,000 Odia examples
Optimized for Odia text generation
Supports various prompting styles in Odia

Training Infrastructure

GPU: Optimized for 16GB+ VRAM
Framework: Unsloth + Transformers
Time: ~2-4 hours training time

Limitations

Trained on limited Odia dataset
May require additional fine-tuning for specific tasks
Best performance with Odia prompts

Citation

If you use this model, please cite:

@misc{odia-mistral-daft,
  title={Odia Mistral 7B DAFT Model},
  author={Your Name},
  year={2025},
  publisher={Hugging Face},
  url={https://huggingface.co/abhilash88/odia-mistral-7b-daft}
}

License

Apache 2.0 License - see LICENSE file for details.

abhilash88/odia-mistral-7b-daft