new vae dataset

mindspore-lab · Feb 11, 2025 · 15f2ca9 · 15f2ca9
1 parent 4a88ab9
commit 15f2ca9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/...ideo/hyvideo/dataset/dataset_videobase.py → ...nyuanvideo/hyvideo/dataset/vae_dataset.py b/...ideo/hyvideo/dataset/dataset_videobase.py → ...nyuanvideo/hyvideo/dataset/vae_dataset.py
diff --git a/examples/hunyuanvideo/scripts/run_vae.py b/examples/hunyuanvideo/scripts/run_vae.py
@@ -19,7 +19,7 @@
 sys.path.append(".")
 
 from hyvideo.constants import PRECISION_TO_TYPE, PRECISIONS, VAE_PATH
-from hyvideo.dataset.dataset_videobase import VideoDataset, create_dataloader
+from hyvideo.dataset.vae_dataset import VideoDataset, create_dataloader
 from hyvideo.utils.data_utils import preprocess_image, preprocess_video, read_video, transform_to_rgb
 from hyvideo.utils.ms_utils import init_env
 from hyvideo.vae import load_vae