Skip to content

Commit

Permalink
[fix] refer use_framewise_encoding on AutoencoderKLHunyuanVideo._enco…
Browse files Browse the repository at this point in the history
…de (#10600)

* fix: refer to use_framewise_encoding on AutoencoderKLHunyuanVideo._encode

* fix: comment about tile_sample_min_num_frames

---------

Co-authored-by: Aryan <[email protected]>
  • Loading branch information
hanchchch and a-r-r-o-w authored Jan 28, 2025
1 parent 658e24e commit f295e2e
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -786,7 +786,7 @@ def __init__(
self.use_tiling = False

# When decoding temporally long video latents, the memory requirement is very high. By decoding latent frames
# at a fixed frame batch size (based on `self.num_latent_frames_batch_sizes`), the memory requirement can be lowered.
# at a fixed frame batch size (based on `self.tile_sample_min_num_frames`), the memory requirement can be lowered.
self.use_framewise_encoding = True
self.use_framewise_decoding = True

Expand Down Expand Up @@ -868,7 +868,7 @@ def disable_slicing(self) -> None:
def _encode(self, x: torch.Tensor) -> torch.Tensor:
batch_size, num_channels, num_frames, height, width = x.shape

if self.use_framewise_decoding and num_frames > self.tile_sample_min_num_frames:
if self.use_framewise_encoding and num_frames > self.tile_sample_min_num_frames:
return self._temporal_tiled_encode(x)

if self.use_tiling and (width > self.tile_sample_min_width or height > self.tile_sample_min_height):
Expand Down

0 comments on commit f295e2e

Please sign in to comment.