More debug printing

gpetters-amd · Sep 11, 2023 · eebc27f · eebc27f
1 parent 479f5f6
commit eebc27f
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -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(