Skip to content

DiffSynth-Studio/examples/wanvideo/model_inference/Wan2.2-Fun-A14B-Control.py 多卡推理出现噪声 #1012

@PrivateMaRyan

Description

@PrivateMaRyan

用这个脚本单卡推理时没有问题,多卡推理时后半段会成为噪声,请问这个是什么原因呢,是不是多卡推理时应当增加什么设置,或者是当前脚本还不支持多卡推理?

推理时设置
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

万望解答,不胜感激!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions