tiny_diff.models.unet.wrappers

Functions

make_fwd_ignore_context(obj)

Wraps object's forward method to ignore the context keyword.