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()