Downloads · 30 days
11
38% of all-time downloads
vanishingradient/mbpp-sft-codeparrot
mbpp-sft-codeparrot is a machine learning model from vanishingradient. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Fine-tuned variant of codeparrot-small on 500 MBPP samples.
Downloads · 30 days
11
38% of all-time downloads
All-time downloads
29
Public
Parameters
111M
444 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors444 MB · 99%
From the Hugging Face model README
Fine-tuned variant of codeparrot-small on 500 MBPP samples.
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))