Remove debug print

KohakuBlueleaf · Jun 22, 2024 · 32f15c9 · 32f15c9
1 parent fb3e8a7
commit 32f15c9
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/hunyuan_train_network.py b/hunyuan_train_network.py
@@ -103,8 +103,6 @@ def get_text_cond(
         ):
             input_ids1 = batch["input_ids"]
             input_ids2 = batch["input_ids2"]
-            print("input_ids1", input_ids1.shape)
-            print("input_ids2", input_ids2.shape)
             with torch.enable_grad():
                 input_ids1 = input_ids1.to(accelerator.device)
                 input_ids2 = input_ids2.to(accelerator.device)
@@ -121,8 +119,6 @@ def get_text_cond(
                         accelerator=accelerator,
                     )
                 )
-                print("encoder_hidden_states1", encoder_hidden_states1.shape)
-                print("encoder_hidden_states2", encoder_hidden_states2.shape)
         else:
             raise NotImplementedError
         return encoder_hidden_states1, mask1, encoder_hidden_states2, mask2