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)
(
self,
x,
time,
cond,
**kwargs,
)
source not stored for this graph (policy: none)