From 3568a057ea4d14afec35795fcd9775514165060c Mon Sep 17 00:00:00 2001 From: TACJu Date: Fri, 24 Jan 2025 16:18:58 -0800 Subject: [PATCH] fix bug and update MaskGen open-source time --- README_MaskGen.md | 3 +-- utils/train_utils.py | 6 +++--- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/README_MaskGen.md b/README_MaskGen.md index f7845b6..948f74c 100644 --- 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)   [![paper](https://img.shields.io/badge/arXiv-Paper-.svg)](http://arxiv.org/abs/2501.07730)   -[![demo](https://img.shields.io/badge/Hugging%20Face-FFD21E?logo=huggingface&logoColor=000)](https://huggingface.co/turkeyju/MaskGen)   @@ -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 index e770ce0..5859dd6 100644 --- 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