zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-20 01:18:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix sequential with empty modules at end (#647)

Browse Source
This commit is contained in:
Awni Hannun
2024-02-07 13:22:27 -08:00
committed by GitHub
parent 28eac18571
commit e5e816a5ef
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_nn.py
View File

@@ -71,7 +71,7 @@ class TestBase(mlx_tests.MLXTestCase):
def test_save_safetensors_weights(self):
def make_model():
return nn.Sequential(nn.Linear(2, 2), nn.ReLU(), nn.Linear(2, 2))
return nn.Sequential(nn.Linear(2, 2), nn.ReLU(), nn.Linear(2, 2), nn.ReLU())
m = make_model()
tdir = tempfile.TemporaryDirectory()