vanishingradient/mbpp-sft-codeparrot

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))