Update pipeline.py (#1)

- Update pipeline.py (93b6af779aa542c60629751f98b3400c3ecc2fa5)


Co-authored-by: Hyoungwon Cho <[email protected]>

pipeline.py

@@ -157,8 +157,8 @@ class PAGIdentitySelfAttnProcessor:
 
         value = attn.to_v(hidden_states_ptb)
         
-        hidden_states_ptb = torch.zeros(value.shape).to(value.get_device())
-        #hidden_states_ptb = value
+        # hidden_states_ptb = torch.zeros(value.shape).to(value.get_device())
+        hidden_states_ptb = value
         
         hidden_states_ptb = hidden_states_ptb.to(query.dtype)