[diffusion] model: support black-forest-labs/FLUX.2-dev (#14000)

2025-11-27 01:49:40 +08:00
parent 0dd759e087
commit 5102d00901
28 changed files with 2685 additions and 175 deletions
@@ -242,6 +242,19 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
            output_size="1024x1024",
        ),
    ),
+    # DiffusionTestCase(
+    #     "flux_2_image_t2i",
+    #     DiffusionServerArgs(
+    #         model_path="black-forest-labs/FLUX.2-dev",
+    #         modality="image",
+    #         warmup_text=1,
+    #         warmup_edit=0,
+    #     ),
+    #     DiffusionSamplingParams(
+    #         prompt="A futuristic cityscape at sunset with flying cars",
+    #         output_size="1024x1024",
+    #     ),
+    # ),
    # === Text and Image to Image (TI2I) ===
    # TODO: Timeout with Torch2.9. Add back when it can pass CI
    # DiffusionTestCase(