added circular VAE Decoder

This commit is contained in:
Lee
2023-08-13 17:09:15 -04:00
parent 5dbf104659
commit 59841afbd8
+15
View File
@@ -100,7 +100,22 @@ class Tiled_KSampler:
return nodes.common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent_image, denoise=denoise)
class CircularVAEDecode:
@classmethod
def INPUT_TYPES(s):
return {"required": { "samples": ("LATENT", ), "vae": ("VAE", )}}
RETURN_TYPES = ("IMAGE",)
FUNCTION = "decode"
CATEGORY = "latent"
def decode(self, vae, samples):
for layer in [layer for layer in vae.first_stage_model.modules() if isinstance(layer, torch.nn.Conv2d)]:
layer.padding_mode = 'circular'
return (vae.decode(samples["samples"]), )
NODE_CLASS_MAPPINGS = {
"Tiled KSampler": Tiled_KSampler,
"Asymmetric Tiled KSampler": Asymmetric_Tiled_KSampler,
"Circular VAEDecode": CircularVAEDecode,
}