cshin23
/

multidim-rm_reg_gating_prototype

Text Classification

pytorch_model_hub_mixin

model_hub_mixin

Model card Files Files and versions Community

cshin23 commited on Jul 26

Commit

a6587b3

•

1 Parent(s): 37f71e0

Update config.json

Files changed (1) hide show

config.json +4 -1

config.json CHANGED Viewed

@@ -1,8 +1,11 @@
 {
   "_name_or_path": "cshin23/multidim-rm_gating_prototype",
   "activation": "gelu",
   "architectures": [
-    "GatedRegressionModel"
   ],
   "attention_dropout": 0.1,
   "dim": 768,

 {
   "_name_or_path": "cshin23/multidim-rm_gating_prototype",
   "activation": "gelu",
+  "auto_map": {
+    "AutoModelForSequenceClassification": "modeling_custom.GatedRegression"
+  },
   "architectures": [
+    "GatedRegression"
   ],
   "attention_dropout": 0.1,
   "dim": 768,