├── .gitignore ├── LICENSE ├── README.md ├── datasets └── .gitkeep ├── examples ├── lb_put_the_red_mug_on_the_left_plate_ep0_demo.mp4 └── lb_tk65_put_the_red_mug_on_the_left_plate_t0_ep49.png ├── flowdiffusion ├── config │ ├── libero_ep20_aug_luotest.py │ └── libero_ep20_bs12_aug.py ├── diffuser │ └── utils │ │ ├── __init__.py │ │ ├── config.py │ │ ├── eval_utils.py │ │ ├── git_utils.py │ │ ├── serialization.py │ │ └── setup.py ├── goal_diffusion.py ├── guided_diffusion │ └── guided_diffusion │ │ ├── dist_util.py │ │ ├── fp16_util.py │ │ ├── gaussian_diffusion.py │ │ ├── image_datasets.py │ │ ├── imagen.py │ │ ├── logger.py │ │ ├── losses.py │ │ ├── nn.py │ │ ├── resample.py │ │ ├── respace.py │ │ ├── script_util.py │ │ ├── train_util.py │ │ └── unet.py ├── libero │ ├── lb_extract_imgs.py │ ├── lb_extract_imgs.sh │ ├── libero_datasets_v2.py │ ├── libero_datasets_v2_sg.py │ ├── plan_libero.sh │ ├── train_libero.py │ └── train_libero.sh ├── model │ ├── __init__.py │ ├── attention_processor.py │ ├── imagen.py │ ├── myunet.py │ ├── resnet.py │ ├── transformer_temporal.py │ ├── unet_3d_blocks.py │ └── unet_3d_condition.py ├── unet.py └── utils.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/README.md -------------------------------------------------------------------------------- /datasets/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/lb_put_the_red_mug_on_the_left_plate_ep0_demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/examples/lb_put_the_red_mug_on_the_left_plate_ep0_demo.mp4 -------------------------------------------------------------------------------- /examples/lb_tk65_put_the_red_mug_on_the_left_plate_t0_ep49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/examples/lb_tk65_put_the_red_mug_on_the_left_plate_t0_ep49.png -------------------------------------------------------------------------------- /flowdiffusion/config/libero_ep20_aug_luotest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/config/libero_ep20_aug_luotest.py -------------------------------------------------------------------------------- /flowdiffusion/config/libero_ep20_bs12_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/config/libero_ep20_bs12_aug.py -------------------------------------------------------------------------------- /flowdiffusion/diffuser/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/diffuser/utils/__init__.py -------------------------------------------------------------------------------- /flowdiffusion/diffuser/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/diffuser/utils/config.py -------------------------------------------------------------------------------- /flowdiffusion/diffuser/utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/diffuser/utils/eval_utils.py -------------------------------------------------------------------------------- /flowdiffusion/diffuser/utils/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/diffuser/utils/git_utils.py -------------------------------------------------------------------------------- /flowdiffusion/diffuser/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/diffuser/utils/serialization.py -------------------------------------------------------------------------------- /flowdiffusion/diffuser/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/diffuser/utils/setup.py -------------------------------------------------------------------------------- /flowdiffusion/goal_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/goal_diffusion.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/dist_util.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/fp16_util.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/image_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/image_datasets.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/imagen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/imagen.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/logger.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/losses.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/nn.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/resample.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/respace.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/script_util.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/train_util.py -------------------------------------------------------------------------------- /flowdiffusion/guided_diffusion/guided_diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/guided_diffusion/guided_diffusion/unet.py -------------------------------------------------------------------------------- /flowdiffusion/libero/lb_extract_imgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/libero/lb_extract_imgs.py -------------------------------------------------------------------------------- /flowdiffusion/libero/lb_extract_imgs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/libero/lb_extract_imgs.sh -------------------------------------------------------------------------------- /flowdiffusion/libero/libero_datasets_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/libero/libero_datasets_v2.py -------------------------------------------------------------------------------- /flowdiffusion/libero/libero_datasets_v2_sg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/libero/libero_datasets_v2_sg.py -------------------------------------------------------------------------------- /flowdiffusion/libero/plan_libero.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/libero/plan_libero.sh -------------------------------------------------------------------------------- /flowdiffusion/libero/train_libero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/libero/train_libero.py -------------------------------------------------------------------------------- /flowdiffusion/libero/train_libero.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/libero/train_libero.sh -------------------------------------------------------------------------------- /flowdiffusion/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/__init__.py -------------------------------------------------------------------------------- /flowdiffusion/model/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/attention_processor.py -------------------------------------------------------------------------------- /flowdiffusion/model/imagen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/imagen.py -------------------------------------------------------------------------------- /flowdiffusion/model/myunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/myunet.py -------------------------------------------------------------------------------- /flowdiffusion/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/resnet.py -------------------------------------------------------------------------------- /flowdiffusion/model/transformer_temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/transformer_temporal.py -------------------------------------------------------------------------------- /flowdiffusion/model/unet_3d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/unet_3d_blocks.py -------------------------------------------------------------------------------- /flowdiffusion/model/unet_3d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/model/unet_3d_condition.py -------------------------------------------------------------------------------- /flowdiffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/unet.py -------------------------------------------------------------------------------- /flowdiffusion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/flowdiffusion/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/video-to-action/v2a-video-model-release/HEAD/requirements.txt --------------------------------------------------------------------------------