disable trt

Dockerfile

@@ -50,7 +50,7 @@ ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
 RUN pip3 install --no-cache-dir --upgrade --pre -r /code/requirements.txt
 RUN pip3 install torch torchvision xformers --index-url https://download.pytorch.org/whl/cu121
 RUN git submodule update --init --recursive
-RUN pip3 install -e ."[tensorrt]"
+RUN pip3 install -e .
 
 # download models`
 RUN mkdir models

demo/start.sh

@@ -8,4 +8,4 @@ else
     echo -e "\033[1;31m\nfrontend build failed\n\033[0m" >&2  exit 1
 fi
 cd ../
-python3 main.py --port 7860 --host 0.0.0.0  --engine-dir ../engines --acceleration tensorrt
+python3 main.py --port 7860 --host 0.0.0.0  --engine-dir ../engines --acceleration none