🔥 Remove print in module code

Signed-off-by: Evaline Ju <69598118+evaline-ju@users.noreply.github.com>
caikit · Jul 28, 2023 · 697c494 · 697c494
1 parent 0ebd730
commit 697c494
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/caikit_nlp/modules/text_generation/peft_prompt_tuning.py b/caikit_nlp/modules/text_generation/peft_prompt_tuning.py
@@ -752,7 +752,6 @@ def get_exportable_prompt_vectors(
         # currently has duplicate values in the second half.
 
         for output_type in output_model_types:
-            print("output_type: ", output_type)
             prompt_dict[output_type.name] = prompt_vector[:num_virtual_tokens]
 
         return prompt_dict