Instructions to use bhargavcn/code-gen-gpt2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use bhargavcn/code-gen-gpt2 with PEFT:
Task type is invalid.
- Transformers
How to use bhargavcn/code-gen-gpt2 with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("bhargavcn/code-gen-gpt2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Configuration Parsing Warning:In adapter_config.json: "peft.task_type" must be a string
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
- PEFT 0.18.1
- Downloads last month
- 8
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for bhargavcn/code-gen-gpt2
Base model
openai-community/gpt2