update CLIP

basetenlabs · Jan 6, 2025 · aa9ced9 · aa9ced9
1 parent d7d181d
commit aa9ced9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/clip/config.yaml b/clip/config.yaml
@@ -4,11 +4,11 @@ model_metadata:
   example_model_input:
     url: https://images.pexels.com/photos/1170986/pexels-photo-1170986.jpeg?auto=compress&cs=tinysrgb&w=1600
 model_name: clip-example
-python_version: py39
+python_version: py311
 requirements:
-- transformers==4.32.0
-- pillow==10.0.0
-- torch==2.0.1
+- transformers==4.47.1
+- pillow
+- torch
 resources:
   accelerator: A10G
   cpu: '3'

diff --git a/clip/model/model.py b/clip/model/model.py
@@ -58,6 +58,7 @@ def preprocess(self, request: Dict) -> Dict:
             padding=True,
         )
         return request
+
 
     # The `predict` method performs the actual inference, and outputs a probability associated
     # with each of the labels defined earlier.