Model reference · open weights
codebert is an open-weight embedding model from microsoft, listed in the AxForge catalogue. AxForge can bring it up on EU-owned hardware for you on request — with the licence handled where one is required.
About
CodeBERT-base Pretrained weights for CodeBERT: A Pre-Trained Model for Programming and Natural Languages. Training Data The model is trained on bi-modal data (documents & code) of CodeSearchNet Training Objective This model is initialized with Roberta-base and trained with MLM+RTD objective (cf. the paper). Usage Please see the official repository for scripts that support "code search" and "code-to-document generation". Reference 1. CodeBERT trained with Masked LM objective (suitable for code completion) 2. 🤗 Hugging Face's CodeBERTa (small size, 6 layers) Citation
Summarised from the published model card. Read the full card on the HuggingFace links below.
Specifications
| Maker | microsoft |
|---|---|
| Type | Embedding models |
| Variants | 1 |
| Runs with | transformers |
| Released | 2022-03-02 |
| Popularity | 202k downloads / month |
| Likes | 287 |
| Licence | Commercial licence needed |
How it works
Variants
Open weights ship in several sizes and precisions. One page, all the variants — pick the one that fits your GPU. VRAM figures are estimates from model size.
| Variant | Params | Precision | VRAM | Fits 16 GB | Weights |
|---|---|---|---|---|---|
| codebert-base | — | BF16 | — | — | Weights ↗ |
Using it via the API
Once AxForge deploys codebert for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (codebert below is illustrative; you get the exact model name on deployment.)
$ curl -sS https://api.axforge.ai/v1/embeddings \
-H "Authorization: Bearer $AXFORGE_API_KEY" \
-H "Content-Type: application/json" \
-d '{"model":"codebert","input":"text to embed"}'
Licence
The weights are open but its licence needs a commercial agreement for business use. AxForge can arrange that licence and host the model for you — you pay AxForge, we settle with the model’s maker. Ask us for a quote. Read the licence ↗
Explore