Fix import error for cfg

Author: Ubiquinone-dot

models/rfd3/src/rfd3/model/inference_sampler.py

@@ -4,6 +4,7 @@
 
 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