How to use from the
Use from the
Transformers library
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("ThoughtFocusAI/CodeGeneration-CodeT5-small")
model = AutoModelForSeq2SeqLM.from_pretrained("ThoughtFocusAI/CodeGeneration-CodeT5-small")
Quick Links

CodeGeneration-CodeT5-base

This model was trained from the pretrained CodeT5-small model on an unknown dataset. It achieves the following results on the evaluation set:

ngram match: 0.004416605133373854
weighted ngram match: 0.0054556977099450635
syntax_match: 0.1444141689373297
dataflow_match: 0.31654676258992803
bleu = 0.91
codebleu = 0.11770830859264417
exact match = 0.0

Model description

This model was trained from the pretrained CodeT5-small model Salesforce and has been fine tuned for Python Code Genertion.

Downloads last month
4
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Space using ThoughtFocusAI/CodeGeneration-CodeT5-small 1