diff --git a/tests/modules/test_per_sample_gradients.py b/tests/modules/test_per_sample_gradients.py index b18e6d9..c4dfa97 100644 --- a/tests/modules/test_per_sample_gradients.py +++ b/tests/modules/test_per_sample_gradients.py @@ -99,7 +99,6 @@ def for_loop_per_sample_gradient( "conv", "conv_bn", "bert", - "gpt", ], ) @pytest.mark.parametrize("use_measurement", [True, False])