Bump version to 0.6.1 (#326)

* Bump version to 0.6.1.

* Re-disable Llama 2 tests.

Author: rmitsch

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]

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():