File size: 1,179 Bytes
c1aff72 82a5e43 c1aff72 82a5e43 c1aff72 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
---
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) |