xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 16:47:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ForcedTiledVAE subclass of VAE

Browse Source
This commit is contained in:
Oscar Anghammar 2025-05-03 17:31:36 +02:00
parent b43916a134
commit 779bbdf942
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
comfy/sd.py
View File

@ -718,6 +718,26 @@ class VAE:
except: except:
return None return None
class ForcedTiledVAE(VAE):
def __init__(self, tile_size, overlap, temporal_size=64, temporal_overlap=8, sd=None, device=None, config=None, dtype=None, metadata=None):
super().__init__(sd, device, config, dtype, metadata)
self.tile_size = tile_size
self.overlap = overlap
self.temporal_size = temporal_size
self.temporal_overlap = temporal_overlap
@classmethod
def cast(cls, vae: VAE, tile_size, overlap, temporal_size=64, temporal_overlap=8):
vae.__class__ = cls
vae.tile_size = tile_size
vae.overlap = overlap
vae.temporal_size = temporal_size
vae.temporal_overlap = temporal_overlap
return vae
def encode(self, pixel_samples):
return super().encode_tiled(pixel_samples, self.tile_size, self.tile_size, self.overlap, self.temporal_size, self.temporal_overlap)
class StyleModel: class StyleModel:
def __init__(self, model, device="cpu"): def __init__(self, model, device="cpu"):
self.model = model self.model = model