multimodalart HF staff hyoungwoncho commited on
Commit
5c21c5c
1 Parent(s): 592e9f9

Update pipeline.py (#1)

Browse files

- Update pipeline.py (93b6af779aa542c60629751f98b3400c3ecc2fa5)


Co-authored-by: Hyoungwon Cho <hyoungwoncho@users.noreply.huggingface.co>

Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -157,8 +157,8 @@ class PAGIdentitySelfAttnProcessor:
157
 
158
  value = attn.to_v(hidden_states_ptb)
159
 
160
- hidden_states_ptb = torch.zeros(value.shape).to(value.get_device())
161
- #hidden_states_ptb = value
162
 
163
  hidden_states_ptb = hidden_states_ptb.to(query.dtype)
164
 
 
157
 
158
  value = attn.to_v(hidden_states_ptb)
159
 
160
+ # hidden_states_ptb = torch.zeros(value.shape).to(value.get_device())
161
+ hidden_states_ptb = value
162
 
163
  hidden_states_ptb = hidden_states_ptb.to(query.dtype)
164