mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-10 18:05:16 +00:00
Improve temporal VAE Encode (Tiled) math.
This commit is contained in:
parent
19a64d6291
commit
ee9547ba31
15
comfy/sd.py
15
comfy/sd.py
@ -572,13 +572,20 @@ class VAE:
|
|||||||
elif dims == 2:
|
elif dims == 2:
|
||||||
samples = self.encode_tiled_(pixel_samples, **args)
|
samples = self.encode_tiled_(pixel_samples, **args)
|
||||||
elif dims == 3:
|
elif dims == 3:
|
||||||
|
if tile_t is not None:
|
||||||
|
tile_t_latent = max(2, self.downscale_ratio[0](tile_t))
|
||||||
|
else:
|
||||||
|
tile_t_latent = 9999
|
||||||
|
args["tile_t"] = self.upscale_ratio[0](tile_t_latent)
|
||||||
|
|
||||||
if overlap_t is None:
|
if overlap_t is None:
|
||||||
args["overlap"] = (1, overlap, overlap)
|
args["overlap"] = (1, overlap, overlap)
|
||||||
else:
|
else:
|
||||||
args["overlap"] = (overlap_t, overlap, overlap)
|
args["overlap"] = (self.upscale_ratio[0](max(1, min(tile_t_latent // 2, self.downscale_ratio[0](overlap_t)))), overlap, overlap)
|
||||||
if tile_t is not None:
|
maximum = pixel_samples.shape[2]
|
||||||
args["tile_t"] = tile_t
|
maximum = self.upscale_ratio[0](self.downscale_ratio[0](maximum))
|
||||||
samples = self.encode_tiled_3d(pixel_samples, **args)
|
|
||||||
|
samples = self.encode_tiled_3d(pixel_samples[:,:,:maximum], **args)
|
||||||
|
|
||||||
return samples
|
return samples
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user