fix: fixed natural_language_processing, regression and similarity_search

BIGmama-technology · Mar 17, 2024 · 007c54f · 007c54f
1 parent 27aef64
commit 007c54f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/hyko_toolkit/models/natural_language_processing/Dockerfile b/hyko_toolkit/models/natural_language_processing/Dockerfile
@@ -1,4 +1,4 @@
-FROM hyko-sdk-tiny:latest
+FROM hyko-sdk-medium:latest
 
 RUN --mount=type=cache,target=/root/.cache/pip pip install transformers[torch]
 RUN --mount=type=cache,target=/root/.cache/pip pip install accelerate 

diff --git a/hyko_toolkit/models/regression/linear_regression/Dockerfile b/hyko_toolkit/models/regression/linear_regression/Dockerfile
@@ -1,4 +1,4 @@
-FROM hyko-sdk-tiny:latest
+FROM hyko-sdk-medium:latest
 
 RUN --mount=type=cache,target=/root/.cache/pip pip install scikit_learn
 

diff --git a/hyko_toolkit/models/similarity_search/E5_faiss/Dockerfile b/hyko_toolkit/models/similarity_search/E5_faiss/Dockerfile
@@ -1,4 +1,4 @@
-FROM hyko-sdk-tiny:latest
+FROM hyko-sdk-medium:latest
 
 WORKDIR /app