klein.cu is a CUDA/GPU port of klein.c (the CPU-only FLUX.2 Klein text-to-image generator by Camenduru). It preserves the original program structure, model architecture, and inference flow while moving all computationally heavy work onto the GPU. -
View it on GitHub