code-search-net/code_search_net
Viewer • Updated • 4.14M • 24.1k • 329
How to use Shuu12121/CodeModernBERT-Owl with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="Shuu12121/CodeModernBERT-Owl") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("Shuu12121/CodeModernBERT-Owl")
model = AutoModelForMaskedLM.from_pretrained("Shuu12121/CodeModernBERT-Owl")旧版のモデルです。こちらのモデルなど最新版をお使いください!!
| パラメータ / Parameter | 値 / Value |
|---|---|
| vocab_size | 50,004 |
| hidden_size | 768 |
| num_hidden_layers | 12 |
| num_attention_heads | 12 |
| intermediate_size | 3,072 |
| max_position_embeddings | 2,048 |
| type_vocab_size | 2 |
| hidden_dropout_prob | 0.1 |
| attention_probs_dropout_prob | 0.1 |
| local_attention_window | 128 |
| rope_theta | 160,000 |
| local_attention_rope_theta | 10,000 |
Apache-2.0