Instructions to use keras/modern_bert_embed_base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- KerasHub
How to use keras/modern_bert_embed_base with KerasHub:
import keras_hub # Load TextClassifier model text_classifier = keras_hub.models.TextClassifier.from_preset( "hf://keras/modern_bert_embed_base", num_classes=2, ) # Fine-tune text_classifier.fit(x=["Thilling adventure!", "Total snoozefest."], y=[1, 0]) # Classify text text_classifier.predict(["Not my cup of tea."])import keras_hub # Create a MaskedLM model task = keras_hub.models.MaskedLM.from_preset("hf://keras/modern_bert_embed_base")import keras_hub # Create a TextEmbedder model task = keras_hub.models.TextEmbedder.from_preset("hf://keras/modern_bert_embed_base")import keras_hub # Create a Backbone model unspecialized for any task backbone = keras_hub.models.Backbone.from_preset("hf://keras/modern_bert_embed_base") - Keras
How to use keras/modern_bert_embed_base with Keras:
# !pip install -U keras tensorflow huggingface_hub # Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here; # "jax" and "torch" also work for computation once TensorFlow is installed. import os os.environ["KERAS_BACKEND"] = "tensorflow" import keras model = keras.saving.load_model("hf://keras/modern_bert_embed_base") - Notebooks
- Google Colab
- Kaggle
Download task.json from keras/modern_bert_embed_base: direct link, hf CLI and curl.
- Browser
- Download file 8.11 kB
-
https://huggingface.co/keras/modern_bert_embed_base/resolve/main/task.json
- Command line
-
hf download hf://keras/modern_bert_embed_base/task.json
-
curl -L -o task.json https://huggingface.co/keras/modern_bert_embed_base/resolve/main/task.json
8.11 kB
| { | |
| "module": "keras_hub.src.models.modern_bert.modern_bert_text_embedder", | |
| "class_name": "ModernBertTextEmbedder", | |
| "config": { | |
| "backbone": { | |
| "module": "keras_hub.src.models.modern_bert.modern_bert_backbone", | |
| "class_name": "ModernBertBackbone", | |
| "config": { | |
| "name": "modern_bert_backbone", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "vocabulary_size": 50368, | |
| "hidden_dim": 768, | |
| "intermediate_dim": 1152, | |
| "num_layers": 22, | |
| "num_heads": 12, | |
| "local_attention_window": 128, | |
| "global_attn_every_n_layers": 3, | |
| "rotary_max_wavelength": 160000.0, | |
| "local_rotary_max_wavelength": 10000.0, | |
| "layer_norm_epsilon": 1e-05 | |
| }, | |
| "registered_name": "keras_hub>ModernBertBackbone" | |
| }, | |
| "preprocessor": { | |
| "module": "keras_hub.src.models.modern_bert.modern_bert_text_embedder_preprocessor", | |
| "class_name": "ModernBertTextEmbedderPreprocessor", | |
| "config": { | |
| "name": "modern_bert_text_embedder_preprocessor", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "tokenizer": { | |
| "module": "keras_hub.src.models.modern_bert.modern_bert_tokenizer", | |
| "class_name": "ModernBertTokenizer", | |
| "config": { | |
| "name": "modern_bert_tokenizer", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "int32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "config_file": "tokenizer.json", | |
| "sequence_length": null, | |
| "add_prefix_space": false, | |
| "unsplittable_tokens": [ | |
| "|||IP_ADDRESS|||", | |
| "<|padding|>", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| " ", | |
| "|||EMAIL_ADDRESS|||", | |
| "|||PHONE_NUMBER|||", | |
| "<|endoftext|>", | |
| "[UNK]", | |
| "[CLS]", | |
| "[SEP]", | |
| "[PAD]", | |
| "[MASK]", | |
| "[unused0]", | |
| "[unused1]", | |
| "[unused2]", | |
| "[unused3]", | |
| "[unused4]", | |
| "[unused5]", | |
| "[unused6]", | |
| "[unused7]", | |
| "[unused8]", | |
| "[unused9]", | |
| "[unused10]", | |
| "[unused11]", | |
| "[unused12]", | |
| "[unused13]", | |
| "[unused14]", | |
| "[unused15]", | |
| "[unused16]", | |
| "[unused17]", | |
| "[unused18]", | |
| "[unused19]", | |
| "[unused20]", | |
| "[unused21]", | |
| "[unused22]", | |
| "[unused23]", | |
| "[unused24]", | |
| "[unused25]", | |
| "[unused26]", | |
| "[unused27]", | |
| "[unused28]", | |
| "[unused29]", | |
| "[unused30]", | |
| "[unused31]", | |
| "[unused32]", | |
| "[unused33]", | |
| "[unused34]", | |
| "[unused35]", | |
| "[unused36]", | |
| "[unused37]", | |
| "[unused38]", | |
| "[unused39]", | |
| "[unused40]", | |
| "[unused41]", | |
| "[unused42]", | |
| "[unused43]", | |
| "[unused44]", | |
| "[unused45]", | |
| "[unused46]", | |
| "[unused47]", | |
| "[unused48]", | |
| "[unused49]", | |
| "[unused50]", | |
| "[unused51]", | |
| "[unused52]", | |
| "[unused53]", | |
| "[unused54]", | |
| "[unused55]", | |
| "[unused56]", | |
| "[unused57]", | |
| "[unused58]", | |
| "[unused59]", | |
| "[unused60]", | |
| "[unused61]", | |
| "[unused62]", | |
| "[unused63]", | |
| "[unused64]", | |
| "[unused65]", | |
| "[unused66]", | |
| "[unused67]", | |
| "[unused68]", | |
| "[unused69]", | |
| "[unused70]", | |
| "[unused71]", | |
| "[unused72]", | |
| "[unused73]", | |
| "[unused74]", | |
| "[unused75]", | |
| "[unused76]", | |
| "[unused77]", | |
| "[unused78]", | |
| "[unused79]", | |
| "[unused80]", | |
| "[unused81]", | |
| "[unused82]" | |
| ] | |
| }, | |
| "registered_name": "keras_hub>ModernBertTokenizer" | |
| }, | |
| "config_file": "preprocessor.json", | |
| "sequence_length": 8192, | |
| "truncate": "round_robin" | |
| }, | |
| "registered_name": "keras_hub>ModernBertTextEmbedderPreprocessor" | |
| }, | |
| "name": "modern_bert_text_embedder", | |
| "pooling_mode": "mean", | |
| "normalize": true | |
| }, | |
| "registered_name": "keras_hub>ModernBertTextEmbedder" | |
| } |