какую нейросеть используете? очень похоже на модель делиберейт или как ее в стейбл диффьюжн.