Spaces:
Runtime error
Runtime error
Update rpc.py
Browse files
rpc.py
CHANGED
|
@@ -148,7 +148,7 @@ class DiffAttention(keras.layers.Layer):
|
|
| 148 |
lambda_full = lambda_1 - lambda_2 + self.lambda_init
|
| 149 |
att = att1 - lambda_full * att2
|
| 150 |
|
| 151 |
-
|
| 152 |
out = out * (1 - self.lambda_init)
|
| 153 |
return out
|
| 154 |
|
|
|
|
| 148 |
lambda_full = lambda_1 - lambda_2 + self.lambda_init
|
| 149 |
att = att1 - lambda_full * att2
|
| 150 |
|
| 151 |
+
out = att @ v
|
| 152 |
out = out * (1 - self.lambda_init)
|
| 153 |
return out
|
| 154 |
|