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
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

ArchitRastogi
/
BGE-Small-LegalEmbeddings-USCode

Sentence Similarity
Transformers
TensorBoard
Safetensors
sentence-transformers
English
bert
feature-extraction
embeddings
legal
USCode
Eval Results (legacy)
text-embeddings-inference
Model card Files Files and versions
xet
Metrics Training metrics Community

Instructions to use ArchitRastogi/BGE-Small-LegalEmbeddings-USCode with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use ArchitRastogi/BGE-Small-LegalEmbeddings-USCode with Transformers:

    # Load model directly
    from transformers import AutoTokenizer, AutoModel
    
    tokenizer = AutoTokenizer.from_pretrained("ArchitRastogi/BGE-Small-LegalEmbeddings-USCode")
    model = AutoModel.from_pretrained("ArchitRastogi/BGE-Small-LegalEmbeddings-USCode")
  • sentence-transformers

    How to use ArchitRastogi/BGE-Small-LegalEmbeddings-USCode with sentence-transformers:

    from sentence_transformers import SentenceTransformer
    
    model = SentenceTransformer("ArchitRastogi/BGE-Small-LegalEmbeddings-USCode")
    
    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]
  • Notebooks
  • Google Colab
  • Kaggle
BGE-Small-LegalEmbeddings-USCode
535 MB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 6 commits
ArchitRastogi's picture
ArchitRastogi
fixed how to use model
aa18202 verified over 1 year ago
  • 1_Pooling
    v1 of model over 1 year ago
  • checkpoint-500
    v1 of model over 1 year ago
  • runs
    v1 of model over 1 year ago
  • .gitattributes
    1.52 kB
    initial commit over 1 year ago
  • README.md
    3.26 kB
    fixed how to use model over 1 year ago
  • config.json
    736 Bytes
    v1 of model over 1 year ago
  • config_sentence_transformers.json
    123 Bytes
    v1 of model over 1 year ago
  • model.safetensors
    133 MB
    xet
    v1 of model over 1 year ago
  • modules.json
    349 Bytes
    v1 of model over 1 year ago
  • sentence_bert_config.json
    53 Bytes
    v1 of model over 1 year ago
  • special_tokens_map.json
    695 Bytes
    v1 of model over 1 year ago
  • tokenizer.json
    711 kB
    v1 of model over 1 year ago
  • tokenizer_config.json
    1.24 kB
    v1 of model over 1 year ago
  • training_args.bin

    Detected Pickle imports (8)

    • "transformers.trainer_utils.IntervalStrategy",
    • "transformers.trainer_utils.HubStrategy",
    • "accelerate.utils.dataclasses.DistributedType",
    • "transformers.training_args.OptimizerNames",
    • "accelerate.state.PartialState",
    • "torch.device",
    • "transformers.trainer_utils.SchedulerType",
    • "FlagEmbedding.baai_general_embedding.finetune.arguments.RetrieverTrainingArguments"

    How to fix it?

    4.86 kB
    xet
    v1 of model over 1 year ago
  • vocab.txt
    232 kB
    v1 of model over 1 year ago