make test_optimized_block forked again

4 months ago · e9c81b2beb
parent a945711f58
commit e9c81b2beb
1 changed files with 0 additions and 1 deletions
--- a/tests/test_optimized_layers.py
+++ b/tests/test_optimized_layers.py
@ -184,7 +184,6 @@ class UnoptimizedWrappedLlamaBlock(LlamaDecoderLayer):


@pytest.mark.parametrize("device", ["cpu", "cuda:0"])
-# @pytest.mark.forked
 def test_optimized_block(device):
    if device == "cuda:0" and not torch.cuda.is_available():
        pytest.skip("CUDA tests can be run only in CUDA-enabled setups")