CoolFace
Modelpublic

bhargavcn/code-gen-gpt2

sourceHugging Faceupdated 7mo agoView on Hugging Face
0likes8downloads
Model Card

Model Card for Code Generation Model - GPT2

Model fine-tuned on a code generation dataset (TokenBender/codeinstructions122kalpacastyle) 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

  • —PEFT 0.18.1