Merge pull request #109 from evaline-ju/rm-prnt

🔥 Remove print in module code
caikit · Jul 28, 2023 · 09f4953 · 09f4953
2 parents 0ebd730 + 697c494
commit 09f4953
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