diff --git a/nodes.py b/nodes.py index 41448bb..8afa198 100644 --- a/nodes.py +++ b/nodes.py @@ -671,7 +671,7 @@ class DavchaEmptyLatentImage: DESCRIPTION = "Create a new batch of empty latent images to be denoised via sampling." def generate(self, width, height, option, batch_size=1): - width, height, upscale_factor = re.findall("(\d+)x(\d+): (\d+\.?\d*)", option)[0] + width, height, upscale_factor = re.findall(r"(\d+)x(\d+): (\d+\.?\d*)", option)[0] width, height, upscale_factor = int(width), int(height), float(upscale_factor) latent = torch.zeros([batch_size, 4, height // 8, width // 8], device=self.device) return ({"samples":latent}, upscale_factor, batch_size)