HF RL Explorer

Implement Krea2Transformer2DModel and Krea2Pipeline for text-to-image generation with the Krea 2 architecture.

Implement Krea2Transformer2DModel and Krea2Pipeline for text-to-image generation with the Krea 2 architecture.: a task in HF ML Tasksmith (Harbor dataset). Krea2Transformer2DModel — a single-stream MMDiT with: - Inputs: packed image latents (B, image seq len, in channels) and stacked text encoder…

The task

**`Krea2Transformer2DModel`** — a single-stream MMDiT with: - Inputs: packed image latents `(B, image_seq_len, in_channels)` and stacked text encoder states `(B, text_seq_len, num_text_layers, text_hidden_dim)`. - A text fusion stage collapses the layer axis: per-token layerwise attention blocks → learned linear…

Part of FineEnvs/HF_ML_Tasksmith.