Add ring/llama4 aux-free adapters and EP sync support

This commit is contained in:
lhl
2025-10-27 14:42:36 +09:00
committed by Wing Lian
parent 3e4688289c
commit 2af7475fdf
7 changed files with 331 additions and 13 deletions

View File

@@ -12,7 +12,10 @@ from pathlib import Path
import torch
from packaging import version
from tbparse import SummaryReader
try:
from tbparse import SummaryReader
except ImportError: # pragma: no cover - optional dependency
SummaryReader = None
from axolotl.utils.dict import DictDefault
@@ -185,6 +188,8 @@ def check_tensorboard(
"""
helper function to parse and check tensorboard logs
"""
if SummaryReader is None:
raise unittest.SkipTest("tbparse is not installed; skipping tensorboard assertions")
tb_log_path = most_recent_subdir(temp_run_dir)
event_file = os.path.join(tb_log_path, sorted(os.listdir(tb_log_path))[0])
reader = SummaryReader(event_file)