[Diffusion] [NPU] [CI] fix CI performance (#19486)
This commit is contained in:
@@ -124,3 +124,8 @@ class NPUPlatformBase(Platform):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def get_device_communicator_cls(cls) -> str:
|
def get_device_communicator_cls(cls) -> str:
|
||||||
return "sglang.multimodal_gen.runtime.distributed.device_communicators.cuda_communicator.CudaCommunicator" # noqa
|
return "sglang.multimodal_gen.runtime.distributed.device_communicators.cuda_communicator.CudaCommunicator" # noqa
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def enable_dit_layerwise_offload_for_wan_by_default(cls) -> bool:
|
||||||
|
"""The performance of the layerwise_offload feature depends on the device's memory size and the memory size occupied by the model. Use --dit-layerwise-offload True if it suitable for your case."""
|
||||||
|
return False
|
||||||
|
|||||||
@@ -12,7 +12,7 @@
|
|||||||
"LatentPreparationStage": 0.25,
|
"LatentPreparationStage": 0.25,
|
||||||
"TimestepPreparationStage": 2.9,
|
"TimestepPreparationStage": 2.9,
|
||||||
"DenoisingStage": 26188.0,
|
"DenoisingStage": 26188.0,
|
||||||
"DecodingStage": 142.03,
|
"DecodingStage": 320.03,
|
||||||
"per_frame_generation": null
|
"per_frame_generation": null
|
||||||
},
|
},
|
||||||
"denoise_step_ms": {
|
"denoise_step_ms": {
|
||||||
|
|||||||
Reference in New Issue
Block a user