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

This commit is contained in:
Mick
2025-11-27 01:49:40 +08:00
committed by GitHub
parent 0dd759e087
commit 5102d00901
28 changed files with 2685 additions and 175 deletions
+10
View File
@@ -26,6 +26,7 @@ from sglang.multimodal_gen.configs.pipeline_configs import (
WanT2V720PConfig,
)
from sglang.multimodal_gen.configs.pipeline_configs.base import PipelineConfig
from sglang.multimodal_gen.configs.pipeline_configs.flux import Flux2PipelineConfig
from sglang.multimodal_gen.configs.pipeline_configs.qwen_image import (
QwenImageEditPipelineConfig,
QwenImagePipelineConfig,
@@ -394,6 +395,15 @@ def _register_configs():
],
model_detectors=[lambda id: "flux" in id.lower()],
)
register_configs(
model_name="flux-2",
sampling_param_cls=FluxSamplingParams,
pipeline_config_cls=Flux2PipelineConfig,
model_paths=[
"black-forest-labs/FLUX.2-dev",
],
model_detectors=[lambda id: "flux.2" in id.lower()],
)
# Qwen-Image
register_configs(