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