Diffusion Policy

[arhanjain]

Wondering if there's a reason diffusion policy hasn't been merged into master?

There's also a few different branches related to diffusion so it's a bit confusing which one is stable & up to date.

[amandlek]

Great question. The other branches offer a preliminary diffusion policy implementation but we would like to clean it up and also run some tests to make sure the implementation is on par with the "official" diffusion policy codebase.

[sarthakdas]

which branch is currently working? which version of diffusers do you have getting this error:

RuntimeError: Failed to import diffusers.models.unets.unet_2d_condition because of the following error (look up to see its traceback):
module 'torch' has no attribute 'float8_e4m3fn'

[amandlek]

You can find more info about the robocasa branch and using it here