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>
- 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 |
-
|
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 |
|