change print

HNR1 · Jun 13, 2023 · 46972d7 · 46972d7
1 parent 8beee9f
commit 46972d7
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tomesd/patch.py b/tomesd/patch.py
@@ -110,9 +110,7 @@ def forward(
             cross_attention_kwargs = cross_attention_kwargs if cross_attention_kwargs is not None else {}
             torch.set_printoptions(precision=10)
             print('norm', norm_hidden_states[0,0,:6])
-            if self.only_cross_attention:
-                print('encoder', encoder_hidden_states[0,0,:6])
-            print('mask', attention_mask[0,0,:6])
+            print('cross', cross_attention_kwargs)
             attn_output = self.attn1(
                 norm_hidden_states,
                 encoder_hidden_states=encoder_hidden_states if self.only_cross_attention else None,