[diffusion] perf: improve black-forest-labs/FLUX.2-dev (#14040)
This commit is contained in:
@@ -242,19 +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",
|
||||
# ),
|
||||
# ),
|
||||
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(
|
||||
|
||||
Reference in New Issue
Block a user