pyramid-flow

0e56f303 · mashun · 0e56f303 · 0e56f303 · 0e56f303 · 0e56f303
Commit 0e56f303 authored Nov 29, 2024 by mashun
20 changed files
--- a/scripts/train_pyramid_flow.sh
+++ b/scripts/train_pyramid_flow.sh
--- a/scripts/train_pyramid_flow_without_ar.sh
+++ b/scripts/train_pyramid_flow_without_ar.sh
--- a/tools/extract_text_features.py
+++ b/tools/extract_text_features.py
--- a/tools/extract_video_vae_latents.py
+++ b/tools/extract_video_vae_latents.py
--- a/train/train_pyramid_flow.py
+++ b/train/train_pyramid_flow.py
--- a/train/train_video_vae.py
+++ b/train/train_video_vae.py
--- a/trainer_misc/__init__.py
+++ b/trainer_misc/__init__.py
--- a/trainer_misc/communicate.py
+++ b/trainer_misc/communicate.py
--- a/trainer_misc/fsdp_trainer.py
+++ b/trainer_misc/fsdp_trainer.py
--- a/trainer_misc/sp_utils.py
+++ b/trainer_misc/sp_utils.py
--- a/trainer_misc/utils.py
+++ b/trainer_misc/utils.py
--- a/trainer_misc/vae_ddp_trainer.py
+++ b/trainer_misc/vae_ddp_trainer.py
--- a/utils.py
+++ b/utils.py
--- a/video_generation_demo.ipynb
+++ b/video_generation_demo.ipynb
--- a/video_vae/__init__.py
+++ b/video_vae/__init__.py
--- a/video_vae/causal_video_vae_wrapper.py
+++ b/video_vae/causal_video_vae_wrapper.py
--- a/video_vae/context_parallel_ops.py
+++ b/video_vae/context_parallel_ops.py
--- a/video_vae/modeling_block.py
+++ b/video_vae/modeling_block.py
--- a/video_vae/modeling_causal_conv.py
+++ b/video_vae/modeling_causal_conv.py
--- a/video_vae/modeling_causal_vae.py
+++ b/video_vae/modeling_causal_vae.py