mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-04-27 11:18:12 -05:00
Fix Module 15 memoization: Add optional mask parameter to MockTransformerBlock forward method
This commit is contained in:
@@ -1281,7 +1281,7 @@ def test_unit_noninvasive_integration():
|
||||
def __init__(self):
|
||||
self.attention = self
|
||||
|
||||
def forward(self, x):
|
||||
def forward(self, x, mask=None):
|
||||
# Simple pass-through for testing
|
||||
return x
|
||||
|
||||
|
||||
Reference in New Issue
Block a user