Update config.json

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/tmp/model",
+  "_name_or_path": "gradientai/v-alpha-tross",
   "architectures": [
     "LlamaForCausalLM"
   ],
@@ -13,6 +13,17 @@
   "intermediate_size": 28672,
   "max_position_embeddings": 4096,
   "model_type": "llama",
+  "neuron": {
+    "auto_cast_type": "bf16",
+    "batch_size": 1,
+    "checkpoint_id": "gradientai/v-alpha-tross",
+    "checkpoint_revision": "6d51ed47fe16e37d819dc52c8c55fd7a7150d2b3",
+    "compiler_type": "neuronx-cc",
+    "compiler_version": "2.12.68.0+4480452af",
+    "num_cores": 24,
+    "sequence_length": 2048,
+    "task": "text-generation"
+  },
   "num_attention_heads": 64,
   "num_hidden_layers": 80,
   "num_key_value_heads": 8,