From eebc27fdbfd9a708ce2b81c820ee80bdb9e25f01 Mon Sep 17 00:00:00 2001 From: George Petterson Date: Mon, 11 Sep 2023 12:26:48 -0400 Subject: [PATCH] More debug printing --- .../src/pipelines/pipeline_shark_stable_diffusion_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py b/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py index 44debbd322..fecad0dc6d 100644 --- a/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py +++ b/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_utils.py @@ -447,6 +447,7 @@ def encode_prompts_weight( "the batch size of `prompt`." ) + print("PROMPT: " + str(prompt)) text_embeddings, uncond_embeddings = get_weighted_text_embeddings( pipe=self, prompt=prompt, @@ -455,6 +456,7 @@ def encode_prompts_weight( else None, max_embeddings_multiples=max_embeddings_multiples, ) + print("EMBEDDINGS: " + str(text_embeddings)) # SHARK: we are not using num_images_per_prompt # bs_embed, seq_len, _ = text_embeddings.shape # text_embeddings = text_embeddings.repeat(