Fix import error for cfg

This commit is contained in:
jbutch
2025-12-05 14:51:37 -08:00
parent 0fb924882a
commit 73d43c50e1

View File

@@ -4,6 +4,7 @@ from typing import Any, Literal
import torch
from jaxtyping import Float
from rfd3.model.cfg_utils import strip_X
from foundry.common import exists
from foundry.utils.ddp import RankedLogger