add model

configuration_tsp.py

@@ -3,6 +3,15 @@ from transformers import PretrainedConfig
 
 class TSPConfig(PretrainedConfig):
     model_type = "tsp"
+    # Manually set mapping to auto models instead of using `register_for_auto_class`
+    # because it can only register the model class that used to execute `push_to_hub` 
+    auto_map = {
+        "AutoModel": "modeling_tsp.TSPModel",
+        "AutoModelForPreTraining": "modeling_tsp.TSPModelForPreTraining",
+        "AutoModelForTokenClassification": "modeling_tsp.TSPModelForTokenClassification",
+        "AutoModelForSequenceClassification": "modeling_tsp.TSPModelForSequenceClassification",
+        "AutoModelForQuestionAnswering": "modeling_tsp.TSPModelForQuestionAnswering",
+    }
 
     def __init__(
         self,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ec68cb7c934eed76350cc01fec44343220347bd3d42c00b5b253f0d8b7ccee6d
+oid sha256:d6fcda66d84b824b881e7b4dca1390b5f2db91d211156994b985138d71dd8ce5
 size 69713927