mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-06 03:57:56 +08:00
- Add dataset implementations for text-to-video and image-to-video - Include bucket sampler for efficient batch processing - Add utility functions for data processing - Create dataset package structure with proper initialization
13 lines
330 B
Python
13 lines
330 B
Python
from .i2v_dataset import I2VDatasetWithResize, I2VDatasetWithBuckets
|
|
from .t2v_dataset import T2VDatasetWithResize, T2VDatasetWithBuckets
|
|
from .bucket_sampler import BucketSampler
|
|
|
|
|
|
__all__ = [
|
|
"I2VDatasetWithResize",
|
|
"I2VDatasetWithBuckets",
|
|
"T2VDatasetWithResize",
|
|
"T2VDatasetWithBuckets",
|
|
"BucketSampler"
|
|
]
|