make sure einops is 0.6.1 or greater and call allow_ops_in_compiled_g…

…raph

make sure einops is 0.6.1 or greater and call allow_ops_in_compiled_g…

…raph

remove einops exts in favor of using pack and unpack from einops 0.6

just set inpaint_images internally with inpaint_videos for now, expos…

…ing an intuitive keyword argument

just do it right

allow for video inpainting with single mask across frames for starters

fix min snr logic, thanks to @justinlovelace

just force train_step to only accept kwargs

add token shift along time for feedforwards in transformer blocks in …

…unet3d

substitute p2 loss weight with min snr loss weight