let user choose his own prompt token count limit

1 files changed, 6 insertions, 0 deletions

diff --git a/modules/processing.py b/modules/processing.py
index f773a30e..d814d5ac 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -123,6 +123,7 @@ class Processed:
         self.index_of_first_image = index_of_first_image

         self.styles = p.styles

         self.job_timestamp = state.job_timestamp

+        self.max_prompt_tokens = opts.max_prompt_tokens

 

         self.eta = p.eta

         self.ddim_discretize = p.ddim_discretize

@@ -141,6 +142,7 @@ class Processed:
         self.all_subseeds = all_subseeds or [self.subseed]

         self.infotexts = infotexts or [info]

 

+

     def js(self):

         obj = {

             "prompt": self.prompt,

@@ -169,6 +171,7 @@ class Processed:
             "infotexts": self.infotexts,

             "styles": self.styles,

             "job_timestamp": self.job_timestamp,

+            "max_prompt_tokens": self.max_prompt_tokens,

         }

 

         return json.dumps(obj)

@@ -266,6 +269,8 @@ def fix_seed(p):
 def create_infotext(p, all_prompts, all_seeds, all_subseeds, comments, iteration=0, position_in_batch=0):

     index = position_in_batch + iteration * p.batch_size

 

+    max_tokens = getattr(p, 'max_prompt_tokens', opts.max_prompt_tokens)

+

     generation_params = {

         "Steps": p.steps,

         "Sampler": sd_samplers.samplers[p.sampler_index].name,

@@ -281,6 +286,7 @@ def create_infotext(p, all_prompts, all_seeds, all_subseeds, comments, iteration
         "Seed resize from": (None if p.seed_resize_from_w == 0 or p.seed_resize_from_h == 0 else f"{p.seed_resize_from_w}x{p.seed_resize_from_h}"),

         "Denoising strength": getattr(p, 'denoising_strength', None),

         "Eta": (None if p.sampler is None or p.sampler.eta == p.sampler.default_eta else p.sampler.eta),

+        "Max tokens": (None if max_tokens == shared.vanilla_max_prompt_tokens else max_tokens)

     }

 

     generation_params.update(p.extra_generation_params)