zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 16:58:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

format

Browse Source
This commit is contained in:
Awni Hannun 2024-12-09 06:50:16 -08:00
parent aa3defcfb0
commit b2b16500fb
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
llms/tests/test_models.py
View File

@ -827,9 +827,7 @@ class TestModels(unittest.TestCase):
rope_theta=10000, rope_theta=10000,
) )
model = exaone.Model(args) model = exaone.Model(args)
self.model_test_runner( self.model_test_runner(model, args.model_type, args.vocab_size, args.num_layers)
model, args.model_type, args.vocab_size, args.num_layers
)
if __name__ == "__main__": if __name__ == "__main__":