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

  • Log In
  • Sign Up

Salesforce
/
SFR-Embedding-Code-2B_R

Feature Extraction
sentence-transformers
Safetensors
Transformers
codexembed2b
code
retrieval
custom_code
Model card Files Files and versions
xet
Community
15

Instructions to use Salesforce/SFR-Embedding-Code-2B_R with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • sentence-transformers

    How to use Salesforce/SFR-Embedding-Code-2B_R with sentence-transformers:

    from sentence_transformers import SentenceTransformer
    
    model = SentenceTransformer("Salesforce/SFR-Embedding-Code-2B_R", trust_remote_code=True)
    
    sentences = [
        "The weather is lovely today.",
        "It's so sunny outside!",
        "He drove to the stadium."
    ]
    embeddings = model.encode(sentences)
    
    similarities = model.similarity(embeddings, embeddings)
    print(similarities.shape)
    # [3, 3]
  • Transformers

    How to use Salesforce/SFR-Embedding-Code-2B_R with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("feature-extraction", model="Salesforce/SFR-Embedding-Code-2B_R", trust_remote_code=True)
    # Load model directly
    from transformers import AutoModel
    model = AutoModel.from_pretrained("Salesforce/SFR-Embedding-Code-2B_R", trust_remote_code=True, dtype="auto")
  • Notebooks
  • Google Colab
  • Kaggle
New discussion
Resources
  • PR & discussions documentation
  • Code of Conduct
  • Hub documentation

Code training data release

#15 opened 8 months ago by
trumancai

May I ask the environment info?

πŸ‘ 1
1
#14 opened about 1 year ago by
itaowe

CodeXEmbed with CoIR in-domain training

πŸ‘ 1
#13 opened about 1 year ago by
urikz

KeyError: 'CodeXEmbedConfig'

1
#12 opened about 1 year ago by
kangkang227

Copying from a non-meta parameter ... is a no-op

#11 opened about 1 year ago by
bevangelista

OutOfMemoryError for large corpus

#10 opened about 1 year ago by
escher-cqse

Reproduce SFR-Embedding-Code model performance on CoIR Leaderboard

2
#9 opened over 1 year ago by
yliu279

Add library_name and pipeline_tag tags

πŸ”₯ 1
1
#5 opened over 1 year ago by
nielsr
Company
TOS Privacy About Careers
Website
Models Datasets Spaces Pricing Docs