Skip to content

Commit a0a4700

Browse files
authored
[AINode] Distinguish uppder and lower case of model id (apache#15847)
1 parent 16ad6c1 commit a0a4700

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

iotdb-core/ainode/ainode/core/manager/model_manager.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,6 @@ def load_model(self, model_id: str, acceleration: bool = False) -> Callable:
101101
"""
102102
Load the model with the given model_id.
103103
"""
104-
model_id = model_id.lower()
105104
logger.info(f"Load model {model_id}")
106105
try:
107106
model = self.model_storage.load_model(model_id, acceleration)
@@ -115,7 +114,6 @@ def save_model(self, model_id: str, model: nn.Module) -> TSStatus:
115114
"""
116115
Save the model using save_pretrained
117116
"""
118-
model_id = model_id.lower()
119117
logger.info(f"Saving model {model_id}")
120118
try:
121119
self.model_storage.save_model(model_id, model)
@@ -152,4 +150,4 @@ def get_built_in_model_type(self, model_id: str) -> BuiltInModelType:
152150
"""
153151
Get the type of the model with the given model_id.
154152
"""
155-
return self.model_storage.get_built_in_model_type(model_id.lower())
153+
return self.model_storage.get_built_in_model_type(model_id)

0 commit comments

Comments
 (0)