How to use from the
Use from the
Keras library
# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"

import keras

model = keras.saving.load_model("hf://samanehs/test_bert")

This is a Bert model uploaded using the KerasNLP library. This model is related to a Classifier task.

Model config:

  • name: bert_backbone
  • trainable: True
  • vocabulary_size: 30522
  • num_layers: 2
  • num_heads: 2
  • hidden_dim: 128
  • intermediate_dim: 512
  • dropout: 0.1
  • max_sequence_length: 512
  • num_segments: 2

This model card has been generated automatically and should be completed by the model author. See Model Cards documentation for more information.

Downloads last month
4
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support