quincyqiang/

nezha-cn-base

import torch
from transformers import BertTokenizer
from nezha import NeZhaModel, NeZhaConfig

text = "今天[MASK]很好，我[MASK]去公园玩。"
tokenizer = BertTokenizer.from_pretrained(
    "quincyqiang/nezha-cn-base"
)
model = NeZhaModel.from_pretrained(
    "quincyqiang/nezha-cn-base"
)

config = NeZhaConfig.from_pretrained(
    "quincyqiang/nezha-cn-base"
)
model.eval()
inputs = tokenizer(text, return_tensors="pt")

with torch.no_grad():
    outputs = model(**inputs)