Upload config.json with huggingface_hub

config.json

@@ -17,17 +17,25 @@
     "auto_cast_type": null,
     "compiler_type": "neuronx-cc",
     "compiler_version": "2.21.18209.0+043b1bf7",
+    "cpu_backend": false,
     "disable_fallback": false,
     "disable_fast_relayout": false,
     "dynamic_batch_size": false,
     "float_dtype": "fp32",
     "inline_weights_to_neff": true,
-    "instance_type": "trn1",
+    "input_names": [
+      "input_ids",
+      "attention_mask",
+      "token_type_ids"
+    ],
     "int_dtype": "int64",
     "model_type": "bert",
     "optlevel": "2",
     "output_attentions": false,
     "output_hidden_states": false,
+    "output_names": [
+      "logits"
+    ],
     "static_batch_size": 1,
     "static_sequence_length": 32,
     "task": "text-classification",