Update ucf101_image_datasets.py

Vchitect / Latte

Latte: Latent Diffusion Transformer for Video Generation.

Apache License 2.0

1.65k stars 177 forks source link

Closed xszheng2020 closed 7 months ago

xszheng2020 commented 7 months ago

fixing an attribute error

maxin-cn commented 7 months ago

The code under if __name__ == '__main__' does not need to be changed.

xszheng2020 commented 7 months ago

The primary goal of this pr is to add

self.frame_data_path = configs.frame_data_path # important

I think the code under if name == 'main' is not changed?