Update pipeline.py (#3)

- Update pipeline.py (e981e5fef5e9eab015d8085280469202eb0d717f)


Co-authored-by: Gyuseong Lee <gses82@users.noreply.huggingface.co>

pipeline.py

@@ -134,8 +134,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)