Hello, great work! I'd like to ask about my self-developed model with a block diffusion architecture similar to sdar. Which parts can be modified to enable the use of accelerated inference engines JetEngine?
I would be most grateful for any suggestions.