语言: 印度尼西亚语
标签:

• 印尼微型RoBERTa
  许可证: MIT
  数据集:
• 维基百科
  示例:
• 文本: "ikiryo adalah hantu dalam mitologi jepang."

印尼微型RoBERTa模型

使用方法

作为掩码语言模型

from transformers import pipeline  

pretrained_name = "akahana/tiny-roberta-indonesia"  

fill_mask = pipeline(  
    "fill-mask",  
    model=pretrained_name,  
    tokenizer=pretrained_name  
)  

fill_mask("ikiryo adalah <mask> hantu dalam mitologi jepang.")  

在PyTorch中进行特征提取

from transformers import RobertaModel, RobertaTokenizerFast  

pretrained_name = "akahana/tiny-roberta-indonesia"  
model = RobertaModel.from_pretrained(pretrained_name)  
tokenizer = RobertaTokenizerFast.from_pretrained(pretrained_name)  

prompt = "ikiryo adalah <mask> hantu dalam mitologi jepang."  
encoded_input = tokenizer(prompt, return_tensors='pt')  
output = model(**encoded_input)