Update config.json

config.json

@@ -1,27 +1,27 @@
 {
   "_name_or_path": "/home/fabrizio/prediction_lab/chronos_example/output/aws-finetuning/202412241542",
   "architectures": [
-    "T5ForConditionalGeneration"
+    "ChronosBoltModelForForecasting"
   ],
   "chronos_config": {
     "context_length": 9,
-    "eos_token_id": 1,
-    "model_type": "seq2seq",
-    "n_special_tokens": 2,
-    "n_tokens": 4096,
-    "num_samples": 86585,
-    "pad_token_id": 0,
-    "prediction_length": 1,
-    "temperature": 1.0,
-    "tokenizer_class": "MeanScaleUniformBins",
-    "tokenizer_kwargs": {
-      "high_limit": 15.0,
-      "low_limit": -15.0
-    },
-    "top_k": 50,
-    "top_p": 1.0,
-    "use_eos_token": true
+    "input_patch_size": 16,
+    "input_patch_stride": 16,
+    "prediction_length": 64,
+    "quantiles": [
+      0.1,
+      0.2,
+      0.3,
+      0.4,
+      0.5,
+      0.6,
+      0.7,
+      0.8,
+      0.9
+    ],
+    "use_reg_token": true
   },
+  "chronos_pipeline_class": "ChronosBoltPipeline",
   "classifier_dropout": 0.0,
   "d_ff": 3072,
   "d_kv": 64,
@@ -46,5 +46,5 @@
   "torch_dtype": "float32",
   "transformers_version": "4.46.3",
   "use_cache": true,
-  "vocab_size": 4096
+  "vocab_size": 2
 }