Update config.json

config.json

@@ -2,22 +2,8 @@
   "architectures": [
     "UtteranceEmbedings"
   ],
-  "attention_probs_dropout_prob": 0.1,
-  "hidden_dropout_prob": 0.1,
-  "hidden_size": 768,
-  "intermediate_size": 3072,
-  "max_edu_length": 128,
-  "max_edus_per_dialog": 100,
-  "max_position_embeddings": 512,
-  "max_speakers": 200,
-  "model_type": "saute",
-  "num_attention_heads": 1,
-  "num_edu_layers": 2,
-  "num_hidden_layers": 1,
-  "num_speaker_embeddings": 512,
-  "num_token_layers": 2,
-  "speaker_embeddings_size": 768,
-  "torch_dtype": "float32",
-  "transformers_version": "4.52.4",
-  "vocab_size": 30522
+  "auto_map" : {
+    "AutoModel" : "saute--model.UtteranceEmbedding",
+    "AutoConfig": "saute--saute_config.SAUTEConfig"
+  }
 }