CompVis / zigma

A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)
https://taohu.me/zigma
Apache License 2.0
271 stars 18 forks source link

Any code and how to change the dataset to webdataset format? #20

Open huangjch526 opened 2 months ago

huangjch526 commented 2 months ago

I want to make the webdataset format of MSCOCO dataset, could you tell me how to make that in your code?

dongzhuoyao commented 2 months ago

You can follow the READMe in this repo, they also use webdataset.

https://github.com/bytedance/1d-tokenizer?tab=readme-ov-file