Запилил небольшой colab для проверки, работает ли (ну и просто было интересно проверить)
Да, действительно, всё это работает... Только на английском. И я не настолько спец по МашОбу и диффузионкам в частности, чтобы понять, что ему еще надо, чтобы понимать русский.
Anyway, можете пользоваться, если у вас есть гугл-аккаунт, знания английского на уровне "забить в гугл-переводчик и скопировать оттудова", и вы не израсходовали свои computation units (скорее всего нет).
Вот результаты по забитым мной данным: