Views
No views yet
1import torch
2from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline
3
4model = AutoModelForMaskedLM.from_pretrained("EQUES/jpharma-bert-large", torch_dtype=torch.bfloat16)
5tokenizer = AutoTokenizer.from_pretrained("EQUES/jpharma-bert-large")
6fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
7
8results = fill_mask("水は化学式で[MASK]2Oです。")
9
10for result in results:
11 print(result)
12# {'score': 0.49609375, 'token': 55, 'token_str': 'H', 'sequence': '水は化学式でH2Oです。'}
13# {'score': 0.11767578125, 'token': 29257, 'token_str': 'Na', 'sequence': '水は化学式でNa2Oです。'}
14# {'score': 0.047607421875, 'token': 61, 'token_str': 'N', 'sequence': '水は化学式でN2Oです。'}
15# {'score': 0.038330078125, 'token': 16966, 'token_str': 'CH', 'sequence': '水は化学式でCH2Oです。'}
16# {'score': 0.0255126953125, 'token': 66, 'token_str': 'S', 'sequence': '水は化学式でS2Oです。'}