Merge pull request #80 from TACJu/main

fix bug and update MaskGen open-source time
bytedance · Jan 25, 2025 · 377fea5 · 377fea5
2 parents d8b782d + 3568a05
commit 377fea5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/README_MaskGen.md b/README_MaskGen.md
@@ -5,7 +5,6 @@
 
 [![Website](https://img.shields.io/badge/Project-Website-87CEEB)](https://tacju.github.io/projects/maskgen.html)&nbsp;&nbsp;
 [![paper](https://img.shields.io/badge/arXiv-Paper-<COLOR>.svg)](http://arxiv.org/abs/2501.07730)&nbsp;&nbsp;
-[![demo](https://img.shields.io/badge/Hugging%20Face-FFD21E?logo=huggingface&logoColor=000)](https://huggingface.co/turkeyju/MaskGen)&nbsp;&nbsp;
 
 </div>
 
@@ -34,7 +33,7 @@ Building on TA-TiTok, we present MaskGen, a versatile text-to-image masked gener
 ## TODO
 
 - [x] Release training code, inference code and checkpoints of TA-TiTok (ETA: Jan 17)
-- [ ] Release training code, inference code and checkpoints of MaskGen (ETA: Jan 24)
+- [ ] Release training code, inference code and checkpoints of MaskGen (ETA: Jan 31)
 
 
 ## TA-TiTok Model Zoo

diff --git a/utils/train_utils.py b/utils/train_utils.py
@@ -289,9 +289,9 @@ def create_dataloader(config, logger, accelerator):
         crop_size=preproc_config.crop_size,
         random_crop=preproc_config.random_crop,
         random_flip=preproc_config.random_flip,
-        dataset_with_class_label=dataset_config.dataset_with_class_label,
-        dataset_with_text_label=dataset_config.dataset_with_text_label,
-        res_ratio_filtering=preproc_config.res_ratio_filtering
+        dataset_with_class_label=dataset_config.get("dataset_with_class_label", "True"),
+        dataset_with_text_label=dataset_config.get("dataset_with_text_label", "False"),
+        res_ratio_filtering=preproc_config.get("res_ratio_filtering", "False"),
     )
     train_dataloader, eval_dataloader = dataset.train_dataloader, dataset.eval_dataloader