bump versions

premAI-io · Nov 3, 2023 · 1142ace · 1142ace
1 parent a8e4707
commit 1142ace
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/ebd-all-minilm/build-aarch64-apple-darwin.sh b/ebd-all-minilm/build-aarch64-apple-darwin.sh
@@ -1,6 +1,6 @@
 #!/usr/bin/env bash
 set -e
-export VERSION=1.0.4
+export VERSION=1.0.5
 
 test -f venv/bin/activate || python -m venv venv
 source venv/bin/activate

diff --git a/ebd-all-minilm/build.sh b/ebd-all-minilm/build.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 set -e
-export VERSION=1.0.4
+export VERSION=1.0.5
 source "$(dirname "${BASH_SOURCE[0]}")/../utils.sh"
 
 build_cpu ghcr.io/premai-io/embeddings-all-minilm-l6-v2-cpu all-MiniLM-L6-v2 ${@:1}

diff --git a/ebd-all-minilm/models.py b/ebd-all-minilm/models.py
@@ -14,8 +14,5 @@ def embeddings(cls, texts):
     @classmethod
     def get_model(cls, model_path):
         if cls.model is None:
-            cls.model = SentenceTransformer(
-                model_path,
-                device=os.getenv("DEVICE", "cpu"),
-            )
+            cls.model = SentenceTransformer(model_path, device=os.getenv("DEVICE", "cpu"))
         return cls.model