---
language:
- en
license: apache-2.0
tags:
- text-generation-inference
- transformers
- unsloth
- mistral
- trl
- python
base_model: unsloth/mistral-7b-bnb-4bit
---

# AI-DocGen Model


- **Developed by:** eltaiefaymen
- **License:** apache-2.0
- **Finetuned from model :** unsloth/mistral-7b-bnb-4bit

# Overview
This model, AI-DocGen, is a finetuned version of the unsloth/mistral-7b-bnb-4bit model. It has been specifically trained to generate docstrings for Python code snippets according to Python's docstring conventions.

# Features
- **Fast Finetuning:** The model was trained 2x faster using Unsloth and Huggingface's TRL library.
- **Optimized for Documentation:** Generates comprehensive and accurate docstrings for Python functions and classes.
- **Efficient Inference:** Supports 4-bit quantization for reduced memory usage and faster inference.

# Training Details
- **Base Model:** unsloth/mistral-7b-bnb-4bit
- **Training Library:** Huggingface's TRL
- **Training Framework:** Unsloth


[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)