mirror of
https://github.com/kvcache-ai/sglang.git
synced 2026-07-11 01:36:58 +00:00
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):
|
||||
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)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user