Matt Brady commited on
Commit
e9822ed
·
verified ·
1 Parent(s): c9727d1

Add POC stub

Browse files
Files changed (5) hide show
  1. 1_Pooling/config.json +9 -0
  2. README.md +62 -0
  3. config.json +22 -0
  4. modules.json +14 -0
  5. tokenizer_config.json +10 -0
1_Pooling/config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 384,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false
9
+ }
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ tags:
5
+ - sentence-transformers
6
+ - feature-extraction
7
+ - embeddings
8
+ - semantic-search
9
+ - contrastive-learning
10
+ library_name: sentence-transformers
11
+ pipeline_tag: feature-extraction
12
+ ---
13
+
14
+ # org-shared-embeddings
15
+
16
+ Shared embedding model for cross-team semantic search over documentation,
17
+ runbooks, and resolved ticket summaries.
18
+
19
+ Hosted under the **organization** namespace for centralized inference endpoint billing.
20
+
21
+ ## Model description
22
+
23
+ | Property | Value |
24
+ |---|---|
25
+ | Base model | `sentence-transformers/all-MiniLM-L6-v2` |
26
+ | Output dimension | 384 |
27
+ | Pooling | mean |
28
+ | Normalization | L2 |
29
+ | Max sequence length | 256 |
30
+
31
+ ## Intended use
32
+
33
+ - Internal doc search (`/v1/search/docs`)
34
+ - Duplicate ticket detection
35
+ - Clustering for QA review sampling
36
+
37
+ ## Usage
38
+
39
+ ```python
40
+ from sentence_transformers import SentenceTransformer
41
+
42
+ model = SentenceTransformer("matt-ts/org-shared-embeddings")
43
+ query = "How do I rotate API keys for the staging environment?"
44
+ doc = "Staging key rotation: open IAM console, select service account..."
45
+ similarity = model.similarity(query, doc)
46
+ print(similarity) # tensor([[0.72]])
47
+ ```
48
+
49
+ ## Deployment
50
+
51
+ | Endpoint | Region | Instance |
52
+ |---|---|---|
53
+ | `embeddings-prod-us` | us-east-1 | `gpu-l4-small` |
54
+ | `embeddings-prod-eu` | eu-west-1 | `gpu-l4-small` |
55
+
56
+ ## Version history
57
+
58
+ | Version | Date | Notes |
59
+ |---|---|---|
60
+ | v1.2.0 | 2026-01-08 | Added runbook corpus (+12k docs) |
61
+ | v1.1.0 | 2025-09-22 | Ticket summary fine-tune |
62
+ | v1.0.0 | 2025-06-01 | Initial MiniLM baseline |
config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": ["BertModel"],
3
+ "model_type": "bert",
4
+ "torch_dtype": "float32",
5
+ "transformers_version": "4.38.2",
6
+ "vocab_size": 30522,
7
+ "hidden_size": 384,
8
+ "num_hidden_layers": 6,
9
+ "num_attention_heads": 12,
10
+ "intermediate_size": 1536,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "attention_probs_dropout_prob": 0.1,
14
+ "max_position_embeddings": 512,
15
+ "type_vocab_size": 2,
16
+ "initializer_range": 0.02,
17
+ "layer_norm_eps": 1e-12,
18
+ "pad_token_id": 0,
19
+ "embedding_dimension": 384,
20
+ "pooling_mode": "mean",
21
+ "normalize_embeddings": true
22
+ }
modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ }
14
+ ]
tokenizer_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_lower_case": true,
3
+ "model_max_length": 256,
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "cls_token": "[CLS]",
7
+ "unk_token": "[UNK]",
8
+ "mask_token": "[MASK]",
9
+ "tokenizer_class": "BertTokenizer"
10
+ }