Skip to content

add swa version #156

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 7 commits into
base: main
Choose a base branch
from
Open

add swa version #156

wants to merge 7 commits into from

Conversation

liubo0902
Copy link
Collaborator

@liubo0902 liubo0902 commented Apr 8, 2025 •
edited
Loading

add swa version, when using 720p 81 frames, inference time may reduce 30%

test videos

WanVideoWrapper_I2V_00002.mp4
00000008.mp4
WanVideoWrapper_I2V_00005.mp4
WanVideoWrapper_I2V_00006.mp4

@liubo0902 liubo0902 requested a review from hkunzhe April 8, 2025 02:51
@nanwanj
Copy link

nanwanj commented Apr 10, 2025

Hi @liubo0902 , we are interested in this enhancement, however, we don't see inference time reduction when using 14B-Control model with examples/wan2.1_fun/predict_v2v_control.py script. Inference time is same when we set swa=TRUE and swa=FALSE in WanTransformer3DModel init function (https://github.com/aigc-apps/VideoX-Fun/pull/156/files#diff-3a0e419208035f56603399b70631762d9db3a6a4f689cb0088d5750ca480c478R622).

Kindly request to provide some context on how to enable this enhancement. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@hkunzhe hkunzhe Awaiting requested review from hkunzhe

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@liubo0902 @nanwanj