Bump version to 0.6.1 (#326)

* Bump version to 0.6.1. * Re-disable Llama 2 tests.
explosion · Oct 13, 2023 · bfd862a · bfd862a
1 parent 89b5b65
commit bfd862a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,5 +1,5 @@
 [metadata]
-version = 0.6.0
+version = 0.6.1
 description = Integrating LLMs into structured NLP pipelines
 author = Explosion
 author_email = [email protected]

diff --git a/spacy_llm/tests/models/test_llama2.py b/spacy_llm/tests/models/test_llama2.py
@@ -37,7 +37,7 @@
 """
 
 
-# @pytest.mark.skip(reason="CI runner needs more GPU memory")
+@pytest.mark.skip(reason="CI runner needs more GPU memory")
 @pytest.mark.gpu
 @pytest.mark.skipif(not has_torch_cuda_gpu, reason="needs GPU & CUDA")
 def test_init():
@@ -52,7 +52,7 @@ def test_init():
     )
 
 
-# @pytest.mark.skip(reason="CI runner needs more GPU memory")
+@pytest.mark.skip(reason="CI runner needs more GPU memory")
 @pytest.mark.gpu
 @pytest.mark.skipif(not has_torch_cuda_gpu, reason="needs GPU & CUDA")
 def test_init_from_config():
@@ -62,7 +62,7 @@ def test_init_from_config():
     torch.cuda.empty_cache()
 
 
-# @pytest.mark.skip(reason="CI runner needs more GPU memory")
+@pytest.mark.skip(reason="CI runner needs more GPU memory")
 @pytest.mark.gpu
 @pytest.mark.skipif(not has_torch_cuda_gpu, reason="needs GPU & CUDA")
 def test_invalid_model():