[CI] Enable EPD CI for EPD architecture enhancements (#26281)

2026-05-25 23:52:58 +08:00
parent 121cc09405
commit 2aa6995308
2 changed files with 4 additions and 1 deletions
@@ -441,6 +441,7 @@ class WaitingImageRequest:
        recv_req: TokenizedGenerateReqInput,
        mm_processor,
        encoder_urls,
+        model_type,
        host_name,
        receive_count,
    ):
@@ -451,6 +452,7 @@ class WaitingImageRequest:
        self.thread = None
        self.mm_processor = mm_processor
        self.encoder_urls = encoder_urls
+        self.model_type = model_type
        self.host_name = host_name
        self.receive_count = receive_count
        self.num_items_assigned = recv_req.num_items_assigned
@@ -926,6 +928,7 @@ class MMReceiverBase(ABC):
                    recv_req=recv_req,
                    mm_processor=self.mm_processor,
                    encoder_urls=self.encode_urls,
+                    model_type=self.model_type,
                    host_name=self.hostname,
                    receive_count=self.tp_size,
                )