Reference the 2nd prompt embed when call()ing the pipeline

bghira · Jul 7, 2023 · b721a58 · b721a58
1 parent 046a15d
commit b721a58
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/discord_tron_client/classes/image_manipulation/prompt_manipulation.py b/discord_tron_client/classes/image_manipulation/prompt_manipulation.py
@@ -23,6 +23,9 @@ def __init__(self, pipeline, device, use_second_encoder: bool = False):
                 device="cuda",
                 use_penultimate_clip_layer=True
             )
+            pipe_tokenizer = self.pipeline.tokenizer
+            pipe_text_encoder = self.pipeline.text_encoder
+
         else:
             # Any other pipeline uses the first tokenizer/encoder.
             pipe_tokenizer = self.pipeline.tokenizer