the Wuerstchen training script!  Training script The training script is also similar to the Text-to-image training guide, but it’s been modified to support Wuerstchen. This guide focuses on the code that is unique to the Wuerstchen training script. The main() function starts by initializing the image encoder - an EfficientNet - in addition to the usual scheduler and tokenizer.   Copied with ContextManagers(deepspeed_zero_init_disabled_context_m