Transformers How to use HuyTran1301/CodeT5p_with_KP_Attention with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, CodeT5WithKeyphrase
tokenizer = AutoTokenizer.from_pretrained("HuyTran1301/CodeT5p_with_KP_Attention")
model = CodeT5WithKeyphrase.from_pretrained("HuyTran1301/CodeT5p_with_KP_Attention", device_map="auto") 
