Skip to content

Commit

Permalink
Update TensorFlow script
Browse files Browse the repository at this point in the history
  • Loading branch information
lutzroeder committed Dec 31, 2024
1 parent 2ed7b36 commit 042b17b
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions tools/tf
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ install() {
echo "tf install"
venv
${python} -m pip install --quiet --upgrade wheel
${python} -m pip install --quiet --upgrade tf-nightly
${python} -m pip install --quiet --upgrade tf-nightly > /dev/null 2>&1 || true
deactivate
}

Expand Down Expand Up @@ -76,7 +76,9 @@ metadata() {
venv
if [[ $(grep -U $'\x0D' ./source/tf-metadata.json) ]]; then crlf=1; else crlf=; fi
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
${python} ./tools/tf_script.py
if pip show "tf-nightly" > /dev/null 2>&1; then
${python} ./tools/tf_script.py
fi
if [[ -n ${crlf} ]]; then
unix2dos --quiet --newfile ./source/tf-metadata.json ./source/tf-metadata.json
fi
Expand Down

0 comments on commit 042b17b

Please # to comment.