diff --git a/MindIE/MultiModal/HunyuanVideo/hyvideo/inference.py b/MindIE/MultiModal/HunyuanVideo/hyvideo/inference.py index 8af0b49829cd23d9820c930d9a3a18f02e6afa2b..986732d2a7f907d4c24dc01dbf139141efb31c07 100644 --- a/MindIE/MultiModal/HunyuanVideo/hyvideo/inference.py +++ b/MindIE/MultiModal/HunyuanVideo/hyvideo/inference.py @@ -15,7 +15,6 @@ from hyvideo.modules import load_model from hyvideo.text_encoder import TextEncoder from hyvideo.utils.data_utils import align_to from hyvideo.modules.posemb_layers import get_nd_rotary_pos_embed -from hyvideo.modules.fp8_optimization import convert_fp8_linear from hyvideo.diffusion.schedulers import FlowMatchDiscreteScheduler from hyvideo.diffusion.pipelines import HunyuanVideoPipeline from hyvideo.utils.parallel_mgr import ( diff --git a/MindIE/MultiModal/HunyuanVideo/hyvideo/vae/parallel_layers.py b/MindIE/MultiModal/HunyuanVideo/hyvideo/vae/parallel_layers.py index f2d59a4a61ebeef245e20e4b55525a2ad207f79e..bd4fc3ffc0a71a3d5c4531be433709bc19489e73 100644 --- a/MindIE/MultiModal/HunyuanVideo/hyvideo/vae/parallel_layers.py +++ b/MindIE/MultiModal/HunyuanVideo/hyvideo/vae/parallel_layers.py @@ -225,7 +225,7 @@ class PatchCausalConv3d(BaseModule): for i in self.module.conv.dilation: if i != 1: raise ValueError(" dilation is not supported in PatchCausalConv3d ") - if self.module.conv.kernel_size[-2] != 3 or self.module.conv.kernel_size[-2] != 1: + if self.module.conv.kernel_size[-2] != 3 and self.module.conv.kernel_size[-2] != 1: raise ValueError(" PatchCausalConv3d only support kernel_size (3, 3, 3) or (1, 1, 1) ") if self.module.conv.stride != (1, 1, 1): raise ValueError(" PatchCausalConv3d only support stride (1, 1, 1) ")