Merge branch 'main' into main-public

intel · Oct 20, 2023 · 7ce85e9 · 7ce85e9
2 parents 85eded7 + 51a8851
commit 7ce85e9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -67,7 +67,7 @@ COPY . /tmp/intel-transfer-learning
 WORKDIR /tmp/intel-transfer-learning
 
 RUN ${PYTHON} setup.py bdist_wheel && \
-    pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu dist/*.whl
+    ${PYTHON} -m pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu dist/*.whl
 
 # TLT target for deployment
 FROM tlt-base as tlt-prod

diff --git a/tensorflow_requirements.txt b/tensorflow_requirements.txt
@@ -4,5 +4,4 @@ numpy==1.23.5
 tensorflow-datasets==4.9.2; python_version=='3.8'
 tensorflow-datasets==4.9.3; python_version>'3.8'
 tensorflow-hub==0.15.0
-tensorflow==2.12.0
 transformers[tensorflow]==4.34.0