don't keep vae tiling enabled forever

2024-08-30 20:14:39 +03:00
parent 9208585442
commit 248428ccf8
1 changed files with 2 additions and 0 deletions
@@ -353,6 +353,8 @@ class CogVideoDecode:
                tile_overlap_factor_height=tile_overlap_factor_height,
                tile_overlap_factor_width=tile_overlap_factor_width,
            )
+        else:
+            vae.disable_tiling()
        latents = latents.to(vae.dtype)
        latents = latents.permute(0, 2, 1, 3, 4)  # [batch_size, num_channels, num_frames, height, width]
        latents = 1 / vae.config.scaling_factor * latents