Skip to content

GitLab

Unverified Commit 1e42d3d3 authored by Musisoul's avatar Musisoul Committed by GitHub
Browse files

Change vae parallel padding size to enhance video quality (#591)

parent 2620398e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lightx2v/models/video_encoders/hf/wan/vae.py
View file @ 1e42d3d3
......@@ -1234,7 +1234,7 @@ class WanVAE:
chunk_h = total_h // world_size_h
chunk_w = total_w // world_size_w
padding_size = 1
padding_size = 2
# Calculate H dimension slice
if cur_rank_h == 0:
......@@ -1317,7 +1317,7 @@ class WanVAE:
chunk_h = total_h // world_size_h
chunk_w = total_w // world_size_w
padding_size = 1
padding_size = 2
# Calculate H dimension slice
if cur_rank_h == 0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment