InstaDeepAI
/

sCellTransformer

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

Yanisadel commited on Mar 19

Commit

3750642

·

verified ·

1 Parent(s): 187fa63

Upload sCT

Files changed (2) hide show

config.json +1 -1
config.py +1 -1

config.json CHANGED Viewed

@@ -18,7 +18,7 @@
   "layer_norm_eps": 1e-05,
   "mask_token_id": 5,
   "max_positions": 20480,
-  "model_type": "sCT",
   "num_cells": 50,
   "num_downsamples": 8,
   "num_hidden_layers_head": 1,

   "layer_norm_eps": 1e-05,
   "mask_token_id": 5,
   "max_positions": 20480,
+  "model_type": "sCellTransformer",
   "num_cells": 50,
   "num_downsamples": 8,
   "num_hidden_layers_head": 1,

config.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import PretrainedConfig
 @dataclass
 class sCTConfig(PretrainedConfig):  # noqa: N801
-    model_type = "sCT"
     def __init__(self, **kwargs):  # type: ignore
         super().__init__()

 @dataclass
 class sCTConfig(PretrainedConfig):  # noqa: N801
+    model_type = "sCellTransformer"
     def __init__(self, **kwargs):  # type: ignore
         super().__init__()