README.md

metadata

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