fix import qwenvl error in RL engine (#12874)
This commit is contained in:
@@ -2714,7 +2714,10 @@ class BumpAllocator:
|
|||||||
def log_info_on_rank0(logger, msg):
|
def log_info_on_rank0(logger, msg):
|
||||||
from sglang.srt.distributed import get_tensor_model_parallel_rank
|
from sglang.srt.distributed import get_tensor_model_parallel_rank
|
||||||
|
|
||||||
if torch.distributed.is_initialized() and get_tensor_model_parallel_rank() == 0:
|
try:
|
||||||
|
if torch.distributed.is_initialized() and get_tensor_model_parallel_rank() == 0:
|
||||||
|
logger.info(msg)
|
||||||
|
except:
|
||||||
logger.info(msg)
|
logger.info(msg)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user