Merge branch 'main' into main-public

Author: ashahba

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

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