[diffusion] fix: offload text encoder model in image encoding stage (#18317)

2026-02-06 22:55:56 +08:00
parent 3d68bd9d9b
commit 79d409f210
1 changed files with 1 additions and 0 deletions
--- a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/image_encoding.py
+++ b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/image_encoding.py
@@ -77,6 +77,7 @@ class ImageEncodingStage(PipelineStage):
        fields = [
            "image_processor",
            "image_encoder",
+            "text_encoder",
        ]
        for field in fields:
            processor = getattr(self, field, None)