[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
@@ -77,6 +77,7 @@ class ImageEncodingStage(PipelineStage):
        fields = [
            "image_processor",
            "image_encoder",
            "text_encoder",
        ]
        for field in fields:
            processor = getattr(self, field, None)