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