File tree Expand file tree Collapse file tree 3 files changed +99
-46
lines changed
Expand file tree Collapse file tree 3 files changed +99
-46
lines changed Original file line number Diff line number Diff line change 1616# under the License.
1717#
1818
19+ from ainode .core .constant import TSStatusCode
20+ from ainode .core .log import Logger
1921from ainode .core .manager .cluster_manager import ClusterManager
2022from ainode .core .manager .inference_manager import InferenceManager
2123from ainode .core .manager .model_manager import ModelManager
3335)
3436from ainode .thrift .common .ttypes import TSStatus
3537
38+ logger = Logger ()
39+
3640
3741class AINodeRPCServiceHandler (IAINodeRPCService .Iface ):
3842 def __init__ (self ):
Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ packages = [
4949python = " >=3.9, <3.13"
5050numpy = " ^1.21.4"
5151pandas = " ^1.3.5"
52- torch = " > =2.2 .0"
52+ torch = " =2.4 .0"
5353pylru = " ^1.2.1"
5454thrift = " >=0.14.0"
5555dynaconf = " ^3.1.11"
@@ -66,6 +66,7 @@ huggingface_hub = "^0.30.1"
6666black = " 25.1.0"
6767isort = " 6.0.1"
6868transformers = " ==4.40.1"
69+ torchmetrics = " >=1.2.0"
6970
7071[tool .poetry .scripts ]
7172ainode = " ainode.core.script:main"
You can’t perform that action at this time.
0 commit comments