Update config.json

config.json

@@ -1,22 +1,12 @@
 {
-  "parameters": {
-    "model_id": "mistralai/Mistral-7B-Instruct-v0.2",
-    "device_map": "auto",
-    "torch_dtype": "bfloat16",
-    "max_new_tokens": 500,
-    "temperature": 0.7,
-    "top_p": 0.95,
-    "top_k": 50,
-    "num_return_sequences": 1,
-    "pad_token_id": 2,
-    "eos_token_id": 2
-  },
-  "environment": {
-    "CUDA_VISIBLE_DEVICES": "0",
-    "TRANSFORMERS_CACHE": "/cache"
-  },
+  "model_id": "mistralai/Mistral-7B-Instruct-v0.2",
+  "runtime": "python",
+  "framework": "pytorch",
+  "python_version": "3.9",
+  "cuda_version": "11.8",
   "hardware": {
     "accelerator": "gpu",
     "gpu_memory": "24GB"
-  }
+  },
+  "allow_override_parameters": true
 }