mgalkin commited on
Commit
eafb8a0
1 Parent(s): f8e61ea

Upload UltraLinkPrediction

Browse files
Files changed (2) hide show
  1. config.json +38 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "UltraLinkPrediction"
4
+ ],
5
+ "entity_model_cfg": {
6
+ "aggregate_func": "sum",
7
+ "hidden_dims": [
8
+ 64,
9
+ 64,
10
+ 64,
11
+ 64,
12
+ 64,
13
+ 64
14
+ ],
15
+ "input_dim": 64,
16
+ "layer_norm": true,
17
+ "message_func": "distmult",
18
+ "short_cut": true
19
+ },
20
+ "model_type": "ultra",
21
+ "relation_model_cfg": {
22
+ "aggregate_func": "sum",
23
+ "hidden_dims": [
24
+ 64,
25
+ 64,
26
+ 64,
27
+ 64,
28
+ 64,
29
+ 64
30
+ ],
31
+ "input_dim": 64,
32
+ "layer_norm": true,
33
+ "message_func": "distmult",
34
+ "short_cut": true
35
+ },
36
+ "torch_dtype": "float32",
37
+ "transformers_version": "4.35.2"
38
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7076f995d05542e72dd8e0162f15cbc4798e5cc1765080d0d1393abb7a206edb
3
+ size 683868