Upload model

config.json

@@ -2,8 +2,13 @@
   "architectures": [
     "TimesformerScrollprizeModel"
   ],
+  "auto_map": {
+    "AutoConfig": "timesformer_config.TimesformerScrollprizeConfig",
+    "AutoModel": "timesformer_model.TimesformerScrollprizeModel"
+  },
   "depth": 8,
   "dim": 512,
+  "model_type": "timesformerscrollprize",
   "n_heads": 6,
   "num_classes": 16,
   "num_frames": 26,

timesformer_config.py

@@ -1,7 +1,7 @@
 from transformers import PretrainedConfig
 
 
-class TimesformerConfig(PretrainedConfig):
+class TimesformerScrollprizeConfig(PretrainedConfig):
     model_type = "timesformerscrollprize"
 
     def __init__(

timesformer_model.py

@@ -1,4 +1,4 @@
-from .timesformer_config import TimesformerConfig
+from .timesformer_config import TimesformerScrollprizeConfig
 from transformers import PreTrainedModel
 import torch
 from timesformer_pytorch import TimeSformer
@@ -8,8 +8,8 @@ from timesformer_pytorch import TimeSformer
 
 
 
-class TimesformerModel(PreTrainedModel):
-    config_class = TimesformerConfig
+class TimesformerScrollprizeModel(PreTrainedModel):
+    config_class = TimesformerScrollprizeConfig
 
     def __init__(self, config):
         super().__init__(config)