Update model.py
Browse files
model.py
CHANGED
@@ -943,7 +943,7 @@ class TextDecoder(nn.Module):
|
|
943 |
# x = self.rotary.apply_rotary(x, freqs)
|
944 |
|
945 |
for block in self.block:
|
946 |
-
x = block(x, xa=None, mask=mask, enc=
|
947 |
|
948 |
for f in order:
|
949 |
if f in enc:
|
|
|
943 |
# x = self.rotary.apply_rotary(x, freqs)
|
944 |
|
945 |
for block in self.block:
|
946 |
+
x = block(x, xa=None, mask=mask, enc=None, layer=layer)
|
947 |
|
948 |
for f in order:
|
949 |
if f in enc:
|