More runpopd updates

2023-06-23 17:22:51 -04:00 · 2023-06-23 17:22:51 -04:00 · 1f67b89682
parent 64cb3713ae
commit 1f67b89682
2 changed files with 1 additions and 5 deletions
--- a/requirements_runpod.txt
+++ b/requirements_runpod.txt
@ -1,4 +1,5 @@
 torch==2.0.1+cu117 torchvision==0.15.2+cu117 --extra-index-url https://download.pytorch.org/whl/cu117 # no_verify leave this to specify not checking this a verification stage
 xformers==0.0.20 bitsandbytes==0.39.1
 accelerate==0.19.0 tensorboard==2.12.1 tensorflow==2.12.0
+tensorrt
 -r requirements.txt
--- a/setup.sh
+++ b/setup.sh
@ -262,11 +262,6 @@ install_python_dependencies() {
      ;;
  esac

-  if [ "$RUNPOD" = true ]; then
-    echo "Installing tenssort."
-    pip install tensorrt
-  fi
-
  # DEBUG ONLY (Update this version number to whatever PyCharm recommends)
  # pip install pydevd-pycharm~=223.8836.43