kplug / kplug_lm_test.py
xusong28
update
2bb0b26
raw history blame
No virus
335 Bytes
# coding=utf-8
# author: xusong <xusong28@jd.com>
# time: 2021/9/17 17:43
#
# from transformers import pipeline
# nlp_fill = pipeline('fill-mask', topk=10)
#
#
# def from_url():
# MODEL_PATH = "http://storage.jd.com/language-models/kplug/huggingface/pytorch_model.bin"
# model = KplugForMaskedLM.from_pretrained(MODEL_PATH)