change timesfm to 2.0

sarahmish · sarahmish · commit 395abc472377 · 2025-10-13T12:58:19.000-04:00
diff --git a/orion/primitives/jsons/orion.primitives.timesfm.TimesFM.json b/orion/primitives/jsons/orion.primitives.timesfm.TimesFM.json
@@ -43,7 +43,7 @@
             },
             "repo_id": {
                 "type": "str", 
-                "default": "google/timesfm-1.0-200m-pytorch"
+                "default": "google/timesfm-2.0-500m-pytorch"
             },
             "batch_size": {
                 "type": "int", 
@@ -59,4 +59,4 @@
             }
         }
     }
-}
+}
diff --git a/orion/primitives/timesfm.py b/orion/primitives/timesfm.py
@@ -55,7 +55,7 @@ class TimesFM:
     def __init__(self,
                  window_size=256,
                  pred_len=1,
-                 repo_id="google/timesfm-1.0-200m-pytorch",
+                 repo_id="google/timesfm-2.0-500m-pytorch",
                  batch_size=32,
                  freq=0,
                  target=0):

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@`
`43`	`43`	`},`
`44`	`44`	`"repo_id": {`
`45`	`45`	`"type": "str",`
`46`		`- "default": "google/timesfm-1.0-200m-pytorch"`
	`46`	`+ "default": "google/timesfm-2.0-500m-pytorch"`
`47`	`47`	`},`
`48`	`48`	`"batch_size": {`
`49`	`49`	`"type": "int",`
`@@ -59,4 +59,4 @@`
`59`	`59`	`}`
`60`	`60`	`}`
`61`	`61`	`}`
`62`		`-}`
	`62`	`+}`