haoosz / ViCo

Official PyTorch codes for the paper: "ViCo: Detail-Preserving Visual Condition for Personalized Text-to-Image Generation"
MIT License
236 stars 15 forks source link

init text on evaluation #12

Open DJLee68 opened 1 year ago

DJLee68 commented 1 year ago

When I do the evaluation, I need to replace the * character with the self.init_text on this line https://github.com/haoosz/ViCo/blob/28c4c9c352a1af0af1581f9cd6a109342fddc7cc/ldm/data/personalized.py#L263 but I get an error saying that TypeError: replace() argument 2 must be str, not None so how should I set the self.init_text?

Thanks in advance!

okaris commented 1 year ago

It would be good to check where you initialise your dataset

https://github.com/haoosz/ViCo/blob/28c4c9c352a1af0af1581f9cd6a109342fddc7cc/scripts/evaluate_model.py#L79

like here and see if you actually set the init text. It might be initialised from the config files

DJLee68 commented 1 year ago

thanks!