This is an automated email from the ASF dual-hosted git repository.
yongzao pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/iotdb.git
The following commit(s) were added to refs/heads/master by this push:
new a0a4700b932 [AINode] Distinguish uppder and lower case of model id
(#15847)
a0a4700b932 is described below
commit a0a4700b932154be8cdeb2dc486d2c6601e9788a
Author: Yongzao <[email protected]>
AuthorDate: Mon Jun 30 20:42:21 2025 +0800
[AINode] Distinguish uppder and lower case of model id (#15847)
---
iotdb-core/ainode/ainode/core/manager/model_manager.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/iotdb-core/ainode/ainode/core/manager/model_manager.py
b/iotdb-core/ainode/ainode/core/manager/model_manager.py
index 46177378035..c63d0d30c97 100644
--- a/iotdb-core/ainode/ainode/core/manager/model_manager.py
+++ b/iotdb-core/ainode/ainode/core/manager/model_manager.py
@@ -101,7 +101,6 @@ class ModelManager:
"""
Load the model with the given model_id.
"""
- model_id = model_id.lower()
logger.info(f"Load model {model_id}")
try:
model = self.model_storage.load_model(model_id, acceleration)
@@ -115,7 +114,6 @@ class ModelManager:
"""
Save the model using save_pretrained
"""
- model_id = model_id.lower()
logger.info(f"Saving model {model_id}")
try:
self.model_storage.save_model(model_id, model)
@@ -152,4 +150,4 @@ class ModelManager:
"""
Get the type of the model with the given model_id.
"""
- return self.model_storage.get_built_in_model_type(model_id.lower())
+ return self.model_storage.get_built_in_model_type(model_id)