File size: 547 Bytes
ce3e976 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | ---
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
```python
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))
|