diff --git a/cls_main/main_cls.py b/cls_main/main_cls.py
index f5c66f6..f184051 100644
--- a/cls_main/main_cls.py
+++ b/cls_main/main_cls.py
@@ -46,7 +46,7 @@ def init_rknn_model(model_path):
         ret = rknn_lite.load_rknn(model_path)
         if ret != 0:
             raise RuntimeError(f'Load model failed: {ret}')
-        ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
+        ret = rknn_lite.init_runtime()
         if ret != 0:
             raise RuntimeError(f'Init runtime failed: {ret}')
         _global_rknn_instance = rknn_lite