x: (B, Ta, Da) time: (B,) or int, diffusion step cond: dict with key state/rgb; more recent obs at the end state: (B, To, Do) rgb: (B, To, C, H, W) TODO long term: more flexible handling of cond
(
self,
x,
time,
cond: dict,
**kwargs,
)
source not stored for this graph (policy: none)