danon6868 commited on
Commit
72b2f75
·
verified ·
1 Parent(s): d6b9cce

Fix non serializable objects in model config.

Browse files
Files changed (2) hide show
  1. config.json +2 -0
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "finetune_seq_embed_model": false,
3
  "fps_num_heads": 4,
4
  "fps_return_weights": true,
@@ -10,6 +11,7 @@
10
  "max_seq_len": 514,
11
  "seq_embed_model_chunk_size": 256,
12
  "seq_embed_model_name": "facebook/esm2_t33_650M_UR50D",
 
13
  "seq_feature_encoder_dropout": 0.2,
14
  "seq_feature_encoder_num_heads": 4,
15
  "stoichiometry_classes_to_use": [
 
1
  {
2
+ "feature_pooling_strategy": "SelfAttentionPooling",
3
  "finetune_seq_embed_model": false,
4
  "fps_num_heads": 4,
5
  "fps_return_weights": true,
 
11
  "max_seq_len": 514,
12
  "seq_embed_model_chunk_size": 256,
13
  "seq_embed_model_name": "facebook/esm2_t33_650M_UR50D",
14
+ "seq_feature_encoder": "GCNConv",
15
  "seq_feature_encoder_dropout": 0.2,
16
  "seq_feature_encoder_num_heads": 4,
17
  "stoichiometry_classes_to_use": [
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:caafb6c8b9e3fcfaa72a06b989ce56de2db4a09f21704cd94feb86b4624db7e2
3
- size 375756485
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11bfe18c867e4c1fb56da1602a347fe141f2913c4e67a79c184b5eebcb461d29
3
+ size 408577541