Model Card for Code Generation Model - GPT2
Model fine-tuned on a code generation dataset (TokenBender/code_instructions_122k_alpaca_style) using PEFT and TRL on Nvidia Tesla GPU.
Model Details
Model Description
Generates code based on the input prompt. The model is fine-tuned on a code generation dataset using PEFT and TRL.
- Developed by: Bhargav C N
- Model type: gpt2
- Language(s) (NLP): English
- License: MIT
- Finetuned from model [optional]: gpt2
Framework versions