How to use aieng-lab/codet5p-770m_comment-type-python with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="aieng-lab/codet5p-770m_comment-type-python")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("aieng-lab/codet5p-770m_comment-type-python") model = AutoModelForSequenceClassification.from_pretrained("aieng-lab/codet5p-770m_comment-type-python", device_map="auto")
