From b2b16500fb71854efefbe19871b3a37ca343203e Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Mon, 9 Dec 2024 06:50:16 -0800 Subject: [PATCH] format --- llms/tests/test_models.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/llms/tests/test_models.py b/llms/tests/test_models.py index a7cb2c78..263ea849 100644 --- a/llms/tests/test_models.py +++ b/llms/tests/test_models.py @@ -827,9 +827,7 @@ class TestModels(unittest.TestCase): rope_theta=10000, ) model = exaone.Model(args) - self.model_test_runner( - model, args.model_type, args.vocab_size, args.num_layers - ) + self.model_test_runner(model, args.model_type, args.vocab_size, args.num_layers) if __name__ == "__main__":