revert useless code updates

CRZbulabula · CRZbulabula · commit 123e63f4c373 · 2025-12-19T10:39:37.000+08:00
diff --git a/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeConcurrentForecastIT.java b/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeConcurrentForecastIT.java
@@ -79,10 +79,6 @@ private static void prepareDataForTableModel() throws SQLException {
   @Test
   public void concurrentGPUForecastTest() throws SQLException, InterruptedException {
     for (AINodeTestUtils.FakeModelInfo modelInfo : BUILTIN_LTSM_MAP.values()) {
-      if (modelInfo.getModelId().equals("chronos2")) {
-        // TODO: @yunbow30944 Support chronos2 model concurrent inference in GPU
-        continue;
-      }
       concurrentGPUForecastTest(modelInfo);
     }
   }
diff --git a/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/model.py b/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/model.py
diff --git a/iotdb-core/ainode/iotdb/ainode/core/model/model_info.py b/iotdb-core/ainode/iotdb/ainode/core/model/model_info.py
@@ -138,7 +138,7 @@ def __repr__(self):
         repo_id="amazon/chronos-2",
         auto_map={
             "AutoConfig": "config.Chronos2ForecastingConfig",
-            "AutoModelForCausalLM": "models.Chronos2Model",
+            "AutoModelForCausalLM": "model.Chronos2Model",
         },
     ),
 }

Original file line number	Diff line number	Diff line change
`@@ -79,10 +79,6 @@ private static void prepareDataForTableModel() throws SQLException {`
`79`	`79`	`@Test`
`80`	`80`	`public void concurrentGPUForecastTest() throws SQLException, InterruptedException {`
`81`	`81`	`for (AINodeTestUtils.FakeModelInfo modelInfo : BUILTIN_LTSM_MAP.values()) {`
`82`		`- if (modelInfo.getModelId().equals("chronos2")) {`
`83`		`- // TODO: @yunbow30944 Support chronos2 model concurrent inference in GPU`
`84`		`- continue;`
`85`		`- }`
`86`	`82`	`concurrentGPUForecastTest(modelInfo);`
`87`	`83`	`}`
`88`	`84`	`}`
Original file line number	Diff line number	Diff line change
`@@ -138,7 +138,7 @@ def __repr__(self):`
`138`	`138`	`repo_id="amazon/chronos-2",`
`139`	`139`	`auto_map={`
`140`	`140`	`"AutoConfig": "config.Chronos2ForecastingConfig",`
`141`		`- "AutoModelForCausalLM": "models.Chronos2Model",`
	`141`	`+ "AutoModelForCausalLM": "model.Chronos2Model",`
`142`	`142`	`},`
`143`	`143`	`),`
`144`	`144`	`}`