Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

mosaicml
/
mosaic-bert-base

Fill-Mask
Transformers
PyTorch
English
bert
custom_code
Model card Files Files and versions
xet
Community
12

Instructions to use mosaicml/mosaic-bert-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use mosaicml/mosaic-bert-base with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("fill-mask", model="mosaicml/mosaic-bert-base", trust_remote_code=True)
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForMaskedLM
    
    tokenizer = AutoTokenizer.from_pretrained("mosaicml/mosaic-bert-base", trust_remote_code=True)
    model = AutoModelForMaskedLM.from_pretrained("mosaicml/mosaic-bert-base", trust_remote_code=True)
  • Notebooks
  • Google Colab
  • Kaggle
mosaic-bert-base
550 MB
Ctrl+K
Ctrl+K
  • 5 contributors
History: 30 commits
daking's picture
daking
kobindra's picture
kobindra
Create LICENSE (#9)
c89bbad verified about 2 years ago
  • .gitattributes
    1.48 kB
    initial commit about 3 years ago
  • LICENSE
    11.3 kB
    Create LICENSE (#9) about 2 years ago
  • README.md
    14.7 kB
    Update README.md over 2 years ago
  • bert_layers.py
    47.3 kB
    Upload BertForMaskedLM about 3 years ago
  • bert_padding.py
    6.26 kB
    Upload BertForMaskedLM about 3 years ago
  • config.json
    827 Bytes
    Change attention_probs_dropout_prob to 0.1 so that FlashAttention/triton dependencies are avoided over 2 years ago
  • configuration_bert.py
    1.02 kB
    Upload BertForMaskedLM about 3 years ago
  • flash_attn_triton.py
    42.7 kB
    Upload BertForMaskedLM about 3 years ago
  • generation_config.json
    90 Bytes
    Upload BertForMaskedLM about 3 years ago
  • pytorch_model.bin

    Detected Pickle imports (3)

    • "torch._utils._rebuild_tensor_v2",
    • "collections.OrderedDict",
    • "torch.FloatStorage"

    What is a pickle import?

    550 MB
    xet
    Upload BertForMaskedLM about 3 years ago