Views
No views yet
xlm-roberta-basepytorch_model.bin, config.json, tokenizer.*, label_schemes.json, modeling_xlmr_multihead.py1from transformers import AutoTokenizer
2from modeling_xlmr_multihead import XLMRMultiHead
3import torch, json
4
5ckpt = "celalkartoglu/xlmr-multihead-tr"
6
7tok = AutoTokenizer.from_pretrained(ckpt)
8model = XLMRMultiHead(n_intent=60, n_ner=111)
9model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
10model.eval()