Upload spaghetti_ai_script.py

spaghetti_ai_script.py

@@ -442,6 +442,8 @@ default_do_not_create_upscaling_preview_images = 0 # 1
 #
 #    Allow Longer Prompts for Stable Diffusion 1.5 Based Models
 #
+#        https://github.com/huggingface/diffusers/issues/2136#issuecomment-1514969011
+#
 
 enable_longer_prompts = 1
 
@@ -3630,30 +3632,25 @@ def create_preview_image (
 #        This is used to get longer prompts for Stable Diffusion 1.5 based
 #        models.
 #
+#        https://github.com/huggingface/diffusers/issues/2136#issuecomment-1514969011
+#
 #####################
 
 def get_pipeline_embeds(
     pipeline,
     prompt,
     negative_prompt,
-    device
+    device,
+    token_length_of_prompt_text,
+    token_length_of_negative_prompt_text
 ):
 
-    """ Get pipeline embeds for prompts bigger than the maxlength of the pipe
-    :param pipeline:
-    :param prompt:
-    :param negative_prompt:
-    :param device:
-    :return:
-    """
     max_length = pipeline.tokenizer.model_max_length
 
-    # simple way to determine length of tokens
-    count_prompt = len(prompt.split(" "))
-    count_negative_prompt = len(negative_prompt.split(" "))
-
     # create the tensor based on which prompt is longer
-    if count_prompt >= count_negative_prompt:
+
+    if token_length_of_prompt_text >= token_length_of_negative_prompt_text:
+
         input_ids = pipeline.tokenizer(prompt, return_tensors="pt", truncation=False).input_ids.to(device)
         shape_max_length = input_ids.shape[-1]
         negative_ids = pipeline.tokenizer(negative_prompt, truncation=False, padding="max_length",
@@ -4043,7 +4040,30 @@ def create_image_function (
     prompt_text_not_used_substring = ""
     negative_prompt_text_not_used_substring = ""
 
-    if allow_longer_prompts_for_sd_1_5_based_models_field_value == 1:
+
+
+
+
+
+
+
+
+
+
+    tokenizer = pipe.tokenizer
+
+    max_token_length_of_model = pipe.tokenizer.model_max_length - number_of_reserved_tokens
+
+    token_length_of_prompt_text = len(tokenizer.tokenize(prompt_text))
+    token_length_of_negative_prompt_text = len(tokenizer.tokenize(negative_prompt_text))
+
+    if (
+        (
+            (token_length_of_prompt_text > max_token_length_of_model) or
+            (token_length_of_negative_prompt_text > max_token_length_of_model)
+        ) and
+        (allow_longer_prompts_for_sd_1_5_based_models_field_value == 1)
+    ):
 
         #    Longer prompts are allowed
 
@@ -4051,7 +4071,9 @@ def create_image_function (
             pipe,
             prompt_text,
             negative_prompt_text,
-            device
+            device,
+            token_length_of_prompt_text,
+            token_length_of_negative_prompt_text
         )
 
         prompt_text_to_use_inside_pipeline = None
@@ -4066,15 +4088,6 @@ def create_image_function (
         prompt_embeds = None
         negative_prompt_embeds = None
 
-        tokenizer = pipe.tokenizer
-
-        max_token_length_of_model = pipe.tokenizer.model_max_length - number_of_reserved_tokens
-
-        token_length_of_prompt_text = len(tokenizer.tokenize(prompt_text))
-        token_length_of_negative_prompt_text = len(tokenizer.tokenize(negative_prompt_text))
-
-
-
         truncated_prompts = 0
 
         partial_prompt_or_negative_prompt_length_too_long_message = ""
@@ -6812,7 +6825,7 @@ with gr.Blocks(
                         model_types_for_longer_prompts_html += " models"
 
                     allow_longer_prompts_for_sd_1_5_based_models_field = gr.Checkbox(
-                        label = "Allow longer prompts for " + model_types_for_longer_prompts_html + " models when not using the refiner or upscaler",
+                        label = "Allow longer prompts for " + model_types_for_longer_prompts_html + " models when not using the refiner or upscaler. (will sometimes fail)",
                         value = default_allow_longer_prompts_for_sd_1_5_based_models_is_selected,
                         interactive = True,
                         container = True,