Upload folder using huggingface_hub

config.json

@@ -1,7 +1,7 @@
 {
   "model_type": "interfuser",
   "architectures": [
-    "Interfuser"
+    "InterfuserModel"
   ],
   "embed_dim": 256,
   "enc_depth": 6,

model.py

@@ -9,6 +9,9 @@ import torch
 from torch import nn, Tensor
 import torch.nn.functional as F
 from torch.nn.parameter import Parameter
+from transformers import AutoConfig, AutoModel
+from transformers import PretrainedConfig, PreTrainedModel
+
 
 try:
     from timm.models.layers import to_2tuple
@@ -1067,3 +1070,7 @@ class Interfuser(nn.Module):
         traffic_feature_with_vel = torch.cat([traffic_feature, velocity], dim=2)
         traffic = self.traffic_pred_head(traffic_feature_with_vel)
         return traffic, waypoints, is_junction, traffic_light_state, stop_sign, traffic_feature
+
+
+AutoConfig.register("interfuser", InterfuserConfig)
+AutoModel.register(InterfuserConfig, InterfuserModel)