用这个脚本单卡推理时没有问题,多卡推理时后半段会成为噪声,请问这个是什么原因呢,是不是多卡推理时应当增加什么设置,或者是当前脚本还不支持多卡推理?
推理时设置
pipe = WanVideoPipeline.from_pretrained( torch_dtype=torch.bfloat16, device="cuda", use_usp=True, #主要加这个更改 model_configs=[ …… ], )
运行时使用
torchrun --nproc_per_node=2 Wan2.2-Fun-A14B-Control.py
万望解答,不胜感激!