|
---
|
|
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) |