Merge pull request #213 from mobiusml/blip_fix

bump transformers to 4.47.0
mobiusml · Dec 6, 2024 · 29fb45d · 29fb45d
2 parents d9bc745 + 5975749
commit 29fb45d
Show file tree

Hide file tree

Showing 6 changed files with 43 additions and 138 deletions.
diff --git a/aana/tests/deployments/test_hf_blip2_deployment.py b/aana/tests/deployments/test_hf_blip2_deployment.py
@@ -32,7 +32,7 @@ class TestHFBlip2Deployment:
     """Test HuggingFace BLIP2 deployment."""
 
     @pytest.mark.asyncio
-    @pytest.mark.parametrize("image_name", ["Starry_Night.jpeg"])
+    @pytest.mark.parametrize("image_name", ["Starry_Night.jpeg", "Seated_Man.png"])
     async def test_methods(self, setup_deployment, image_name):
         """Test HuggingFace BLIP2 methods."""
         deployment_name, handle_name, _ = setup_deployment

diff --git a/aana/tests/deployments/test_hf_pipeline_deployment.py b/aana/tests/deployments/test_hf_pipeline_deployment.py
@@ -20,6 +20,7 @@
             ray_actor_options={"num_gpus": 1},
             user_config=HfPipelineConfig(
                 model_id="Salesforce/blip2-opt-2.7b",
+                task = "image-to-text",
                 model_kwargs={
                     "quantization_config": BitsAndBytesConfig(
                         load_in_8bit=False, load_in_4bit=True

diff --git a/aana/tests/files/expected/hf_blip2/blip2_deployment_Seated_Man.png.json b/aana/tests/files/expected/hf_blip2/blip2_deployment_Seated_Man.png.json
@@ -0,0 +1 @@
+"a man sitting at a table in front of a plant"
diff --git a/aana/tests/files/images/Seated_Man.png b/aana/tests/files/images/Seated_Man.png