[CI] Enable EPD CI for EPD architecture enhancements (#26281)
This commit is contained in:
@@ -441,6 +441,7 @@ class WaitingImageRequest:
|
|||||||
recv_req: TokenizedGenerateReqInput,
|
recv_req: TokenizedGenerateReqInput,
|
||||||
mm_processor,
|
mm_processor,
|
||||||
encoder_urls,
|
encoder_urls,
|
||||||
|
model_type,
|
||||||
host_name,
|
host_name,
|
||||||
receive_count,
|
receive_count,
|
||||||
):
|
):
|
||||||
@@ -451,6 +452,7 @@ class WaitingImageRequest:
|
|||||||
self.thread = None
|
self.thread = None
|
||||||
self.mm_processor = mm_processor
|
self.mm_processor = mm_processor
|
||||||
self.encoder_urls = encoder_urls
|
self.encoder_urls = encoder_urls
|
||||||
|
self.model_type = model_type
|
||||||
self.host_name = host_name
|
self.host_name = host_name
|
||||||
self.receive_count = receive_count
|
self.receive_count = receive_count
|
||||||
self.num_items_assigned = recv_req.num_items_assigned
|
self.num_items_assigned = recv_req.num_items_assigned
|
||||||
@@ -926,6 +928,7 @@ class MMReceiverBase(ABC):
|
|||||||
recv_req=recv_req,
|
recv_req=recv_req,
|
||||||
mm_processor=self.mm_processor,
|
mm_processor=self.mm_processor,
|
||||||
encoder_urls=self.encode_urls,
|
encoder_urls=self.encode_urls,
|
||||||
|
model_type=self.model_type,
|
||||||
host_name=self.hostname,
|
host_name=self.hostname,
|
||||||
receive_count=self.tp_size,
|
receive_count=self.tp_size,
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -36,7 +36,7 @@ DEFAULT_OMNI_MODEL = "Qwen/Qwen3-Omni-30B-A3B-Instruct"
|
|||||||
QWEN35_27B_MODEL = "Qwen/Qwen3.5-27B"
|
QWEN35_27B_MODEL = "Qwen/Qwen3.5-27B"
|
||||||
|
|
||||||
|
|
||||||
register_cuda_ci(est_time=97, suite="nightly-4-gpu", nightly=True)
|
register_cuda_ci(est_time=97, stage="base-c", runner_config="4-gpu-h100")
|
||||||
|
|
||||||
|
|
||||||
@unittest.skipIf(
|
@unittest.skipIf(
|
||||||
|
|||||||
Reference in New Issue
Block a user