Update tensorflow so we have arm64 wheels

Also downgrade protobuf to meet tensorflow's deps
exadel-inc · Jan 23, 2023 · 43b4953 · 43b4953
1 parent 307d923
commit 43b4953
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/embedding-calculator/src/services/facescan/plugins/agegender/__init__.py b/embedding-calculator/src/services/facescan/plugins/agegender/__init__.py
@@ -12,4 +12,4 @@
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
 
-requirements = ('tensorflow~=2.1.4', 'tf-slim~=1.1.0')
+requirements = ('tensorflow~=2.11.0', 'tf-slim~=1.1.0')
diff --git a/embedding-calculator/src/services/facescan/plugins/dependencies.py b/embedding-calculator/src/services/facescan/plugins/dependencies.py
@@ -18,7 +18,7 @@
 from src.services.utils.pyutils import get_env
 
 
-def get_tensorflow(version='2.1.4') -> Tuple[str, ...]:
+def get_tensorflow(version='2.11.0') -> Tuple[str, ...]:
     libs = [f'tensorflow=={version}']
     cuda_version = get_env('CUDA', '').replace('.', '')
     if ENV.GPU_IDX > -1 and cuda_version:

diff --git a/embedding-calculator/src/services/facescan/plugins/facenet/__init__.py b/embedding-calculator/src/services/facescan/plugins/facenet/__init__.py
@@ -14,4 +14,4 @@
 
 from src.services.facescan.plugins.dependencies import get_tensorflow
 
-requirements = ('protobuf~=3.20.1',) + get_tensorflow() # + ('mtcnn~=0.1.0',)
+requirements = ('protobuf~=3.19.6',) + get_tensorflow() # + ('mtcnn~=0.1.0',)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -14,4 +14,4 @@

		from src.services.facescan.plugins.dependencies import get_tensorflow

		requirements = ('protobuf~=3.20.1',) + get_tensorflow() # + ('mtcnn~=0.1.0',)
		requirements = ('protobuf~=3.19.6',) + get_tensorflow() # + ('mtcnn~=0.1.0',)