wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-30 13:33:42 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
Yousef Rafat 2026-03-19 21:32:23 +02:00
parent af6c5d6de9
commit 3b418dab2c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/seedvr/vae.py
View File

@ -2060,6 +2060,7 @@ class VideoAutoencoderKLWrapper(VideoAutoencoderKL):
self.temporal_downsample_factor = temporal_downsample_factor self.temporal_downsample_factor = temporal_downsample_factor
self.freeze_encoder = freeze_encoder self.freeze_encoder = freeze_encoder
self.original_image_video = None self.original_image_video = None
self.enable_tiling = False
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.set_memory_limit(0.5, 0.5) self.set_memory_limit(0.5, 0.5)
@ -2092,8 +2093,7 @@ class VideoAutoencoderKLWrapper(VideoAutoencoderKL):
if latent.ndim == 4: if latent.ndim == 4:
latent = latent.unsqueeze(2) latent = latent.unsqueeze(2)
if self.tiled_args.get("enable_tiling", None) is not None: self.enable_tiling = self.tiled_args.get("enable_tiling", False)
self.enable_tiling = self.tiled_args.pop("enable_tiling", False)
if self.enable_tiling: if self.enable_tiling:
x = tiled_vae(latent, self, **self.tiled_args, encode=False).squeeze(2) x = tiled_vae(latent, self, **self.tiled_args, encode=False).squeeze(2)