ziffir commited on
Commit
4beb57b
·
verified ·
1 Parent(s): a81c94f

Create demo_data.py

Browse files
Files changed (1) hide show
  1. demo_data.py +21 -0
demo_data.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Phoenix/phoenix_config.yaml
2
+ model:
3
+ architecture: "grok-1-transformer"
4
+ embedding_size: 128
5
+ num_layers: 12
6
+ num_heads: 8
7
+
8
+ retrieval:
9
+ tower_type: "two_tower"
10
+ embedding_dim: 256
11
+ similarity_metric: "cosine"
12
+
13
+ ranking:
14
+ transformer_layers: 6
15
+ attention_heads: 4
16
+ dropout_rate: 0.1
17
+
18
+ optimization:
19
+ use_hashing: true
20
+ hash_table_size: 10000
21
+ cache_embeddings: true