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
Show 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: ...@@ -1234,7 +1234,7 @@ class WanVAE:
chunk_h = total_h // world_size_h chunk_h = total_h // world_size_h
chunk_w = total_w // world_size_w chunk_w = total_w // world_size_w
padding_size = 1 padding_size = 2
# Calculate H dimension slice # Calculate H dimension slice
if cur_rank_h == 0: if cur_rank_h == 0:
...@@ -1317,7 +1317,7 @@ class WanVAE: ...@@ -1317,7 +1317,7 @@ class WanVAE:
chunk_h = total_h // world_size_h chunk_h = total_h // world_size_h
chunk_w = total_w // world_size_w chunk_w = total_w // world_size_w
padding_size = 1 padding_size = 2
# Calculate H dimension slice # Calculate H dimension slice
if cur_rank_h == 0: 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