fix

Author: hiworldwzj

lightllm/utils/envs_utils.py

@@ -234,7 +234,7 @@ def enable_huge_page():
 def get_added_mtp_kv_layer_num() -> int:
     # mtp 模式下需要在mem manger上扩展draft model使用的layer
     added_mtp_layer_num = 0
-    if get_env_start_args().mtp_mode in ["eagle_with_att"]:
+    if get_env_start_args().mtp_mode == "eagle_with_att":
         added_mtp_layer_num += 1
     elif get_env_start_args().mtp_mode == "vanilla_with_att":
         added_mtp_layer_num += get_env_start_args().mtp_step

lightllm/utils/log_utils.py

@@ -115,8 +115,3 @@ def log_time_ready(mark_name, time_count: int):
         return True
     else:
         return False
-
-# print_rank0 use the same as print, but only print on rank 0 in distributed setting
-def print_rank0(*args, **kwargs):
-    if int(os.environ.get("LIGHTLLM_GLOBAL_RANK", "0")) == 0:
-        print(*args, **kwargs)