From 8f393b64b890f8321f5b946af1114af584b069c6 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 24 May 2023 15:00:32 +1000 Subject: [PATCH] feat(nodes): add seed validator If `seed>SEED_MAX`, we can still continue if we parse the seed as `seed % SEED_MAX`. --- invokeai/app/invocations/latent.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py index adba88274f..40ba67861a 100644 --- a/invokeai/app/invocations/latent.py +++ b/invokeai/app/invocations/latent.py @@ -140,6 +140,11 @@ class NoiseInvocation(BaseInvocation): }, } + @validator("seed", pre=True) + def modulo_seed(cls, v): + """Returns the seed modulo SEED_MAX to ensure it is within the valid range.""" + return v % SEED_MAX + def invoke(self, context: InvocationContext) -> NoiseOutput: device = torch.device(choose_torch_device()) noise = get_noise(self.width, self.height, device, self.seed)