pedrocas15 commited on
Commit
291afac
·
verified ·
1 Parent(s): fd69507

Update rpc.py

Browse files
Files changed (1) hide show
  1. rpc.py +1 -1
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
- outi = att @ v
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