Views
No views yet
fill-mask タスクとして直接利用できるほか、
ファインチューニングを行うことで意味的コード検索やコード理解タスクの
ベースモデルとして利用できます。1from transformers import AutoTokenizer, AutoModelForMaskedLM
2#MLMモデルとして読み込む場合
3tokenizer = AutoTokenizer.from_pretrained("Shuu12121/Owl-ph1-base-len512")
4model = AutoModelForMaskedLM.from_pretrained("Shuu12121/Owl-ph1-base-len512")1from sentence_transformers import SentenceTransformer,models
2
3word_embedding_model = models.Transformer("Shuu12121/Owl-ph1-base-len512")
4word_embedding_model.max_seq_length = 512
5#CLSトークンを用いる場合pooling_mode_cls_tokenのみTrue
6#平均プーリングを用いる場合 pooling_mode_mean_tokensのみTrue
7pooling_model = models.Pooling(
8 word_embedding_model.get_word_embedding_dimension(),
9 pooling_mode_cls_token=True,
10 pooling_mode_mean_tokens=False,
11 pooling_mode_max_tokens=False,
12 pooling_mode_weightedmean_tokens=False,
13 pooling_mode_lasttoken=False,
14)
15model = SentenceTransformer(modules=[word_embedding_model, pooling_model])