From dfe7eb8aa401a77bd4e52c410c77d58336afb89b Mon Sep 17 00:00:00 2001 From: asagi4 <130366179+asagi4@users.noreply.github.com> Date: Sat, 7 Dec 2024 18:59:48 +0200 Subject: [PATCH] Don't error out on long prompts --- __init__.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/__init__.py b/__init__.py index 965bebf..32f6171 100644 --- a/__init__.py +++ b/__init__.py @@ -217,9 +217,11 @@ class NPNetGoldenNoise: input_latent = input_latent.copy() print("Latent must be 128x128 for the NPNet model to work; generating square noise and reshaping...") input_latent["samples"] = common_upscale(input_latent["samples"], 128, 128, "nearest-exact", "disabled") - init_noise = self.noise.generate_noise(input_latent).to("cuda") - cond = self.cond[0].clone().to("cuda") - self.npnet.to("cuda") + init_noise = self.noise.generate_noise(input_latent).to(self.npnet.device) + cond = self.cond[0].clone().to(self.npnet.device) + if cond.shape[1] != 77: + print("NPNet can't handle conds >77 tokens, truncating...") + cond = cond[:, :77, :] try: print("Applying NPNet to noise") r = self.npnet(init_noise, cond).to("cpu")