From c62fc6e1bce955084454d3a7f8d07aeb1f3703b4 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 1 Nov 2023 15:55:59 -0700 Subject: [PATCH] speedup --- fooocus_version.py | 2 +- modules/expansion.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/fooocus_version.py b/fooocus_version.py index 0bbac3cd..80f1f10e 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.1.773' +version = '2.1.774' diff --git a/modules/expansion.py b/modules/expansion.py index e26f7229..64d3f07e 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -75,9 +75,12 @@ class FooocusExpansion: @torch.inference_mode() def logits_processor(self, input_ids, scores): assert scores.ndim == 2 and scores.shape[0] == 1 - bias = self.logits_bias.to(scores).clone() + self.logits_bias = self.logits_bias.to(scores) + + bias = self.logits_bias.clone() bias[0, input_ids[0].to(bias.device).long()] = neg_inf bias[0, 11] = 0 + return scores + bias @torch.no_grad()