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
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for eltaiefaymen/AI-DocGen
Base model
unsloth/mistral-7b-bnb-4bit
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("eltaiefaymen/AI-DocGen", dtype="auto")