Use with other pretrained models

[stobermeier]

Is it possible to run the tool with other pretrained models? This repository provides a list of several other pretrained models. However, when trying to use the tool with CelebA-HQ, I run into the error message "ModuleNotFoundError: No module named 'transformers'".

Thank you for your great work!

[jonnm4]

I get a rainbow square that never changes; iteration to iteration staying the same.