Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Hardware
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

ibm-granite
/
granite-embedding-125m-english

Sentence Similarity
sentence-transformers
PyTorch
ONNX
Safetensors
Transformers
English
roberta
feature-extraction
language
granite
embeddings
mteb
Eval Results (legacy)
Eval Results
text-embeddings-inference
Model card Files Files and versions
xet
Community
4

Instructions to use ibm-granite/granite-embedding-125m-english with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • sentence-transformers

    How to use ibm-granite/granite-embedding-125m-english with sentence-transformers:

    from sentence_transformers import SentenceTransformer
    
    model = SentenceTransformer("ibm-granite/granite-embedding-125m-english")
    
    sentences = [
        "That is a happy person",
        "That is a happy dog",
        "That is a very happy person",
        "Today is a sunny day"
    ]
    embeddings = model.encode(sentences)
    
    similarities = model.similarity(embeddings, embeddings)
    print(similarities.shape)
    # [4, 4]
  • Transformers

    How to use ibm-granite/granite-embedding-125m-english with Transformers:

    # Load model directly
    from transformers import AutoTokenizer, AutoModel
    
    tokenizer = AutoTokenizer.from_pretrained("ibm-granite/granite-embedding-125m-english")
    model = AutoModel.from_pretrained("ibm-granite/granite-embedding-125m-english", device_map="auto")
  • Inference
  • Notebooks
  • Google Colab
  • Kaggle
granite-embedding-125m-english
1 GB
Ctrl+K
Ctrl+K
  • 3 contributors
History: 11 commits
pawasthy's picture
pawasthy
Upload 2 files
4ab61ff verified 12 months ago
  • 1_Pooling
    Upload 10 files over 1 year ago
  • .gitattributes
    1.52 kB
    initial commit over 1 year ago
  • README.md
    466 kB
    Upload 2 files 12 months ago
  • config.json
    720 Bytes
    Upload 10 files over 1 year ago
  • model.onnx
    499 MB
    xet
    Upload 2 files 12 months ago
  • model.safetensors
    249 MB
    xet
    Upload model.safetensors over 1 year ago
  • modules.json
    350 Bytes
    Upload 10 files over 1 year ago
  • pytorch_model.bin

    Detected Pickle imports (3)

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

    What is a pickle import?

    249 MB
    xet
    Upload 10 files over 1 year ago
  • sentence_bert_config.json
    54 Bytes
    Upload 10 files over 1 year ago
  • special_tokens_map.json
    957 Bytes
    Upload 10 files over 1 year ago
  • tokenizer.json
    2.11 MB
    Upload 10 files over 1 year ago
  • tokenizer_config.json
    1.32 kB
    Upload 10 files over 1 year ago
  • trainer_state.json
    451 kB
    Upload 10 files over 1 year ago
  • vocab.json
    798 kB
    Upload 10 files over 1 year ago