From ee5ff20245119a39c4ca99b2a729df611aedfba0 Mon Sep 17 00:00:00 2001 From: Sasha Doubov Date: Thu, 30 Nov 2023 23:17:24 +0000 Subject: [PATCH] fix spacing again --- megablocks/layers/glu_test.py | 1 - 1 file changed, 1 deletion(-) diff --git a/megablocks/layers/glu_test.py b/megablocks/layers/glu_test.py index 60759138..5d7319da 100644 --- a/megablocks/layers/glu_test.py +++ b/megablocks/layers/glu_test.py @@ -49,7 +49,6 @@ class GLUTest(parameterized.TestCase): @parameterized.parameters(*_DENSE_TESTS) def testGLU_ForwardGroupedMLP(self, bs, sl, hs): - x = torch.randn(sl, bs, hs).to(torch.bfloat16).cuda() _, glu, dmoe_glu = test_modules(