Downloads · 30 days
0
AI-DocGen
eltaiefaymen/AI-DocGen
AI-DocGen is a machine learning model from eltaiefaymen. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
- Developed by: eltaiefaymen - License: apache-2.0 - Finetuned from model : unsloth/mistral-7b-bnb-4bit
Access
Public
Updated May 26, 2024
Repo size
168 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors168 MB · 99%
At a glance
- Library
- transformers
- License
- apache-2.0
- Access
- Public
- Created
- May 25, 2024
- Updated
- May 26, 2024
- SHA
- c1aff722
Base models
Model card
From the Hugging Face model README
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
- Library
- transformers
- License
- apache-2.0
- Languages
- en
- Created
- May 25, 2024
- Updated
- May 26, 2024