metadata
license: apache-2.0
language: en
tags:
- code-generation
model_type: gpt2
MBPP SFT CodeParrot
Fine-tuned variant of codeparrot-small on 500 MBPP samples.
Usage
from transformers import AutoTokenizer, AutoModelForCausalLM
m = "vanishingradient/mbpp-sft-codeparrot"
tok = AutoTokenizer.from_pretrained(m)
mod = AutoModelForCausalLM.from_pretrained(m)
x = tok("Write a python function to reverse a string.", return_tensors="pt")
y = mod.generate(**x, max_new_tokens=120)
print(tok.decode(y[0], skip_special_tokens=True))