Spaces:
Sleeping
Sleeping
leoxing1996
commited on
Commit
·
468e13e
1
Parent(s):
c3b5900
install pt
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -45,6 +45,7 @@ COPY --chown=user . $HOME/app
|
|
| 45 |
ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
|
| 46 |
|
| 47 |
# install dependencies
|
|
|
|
| 48 |
RUN git submodule update --init --recursive
|
| 49 |
RUN pip install -e ."[tensorrt]"
|
| 50 |
|
|
|
|
| 45 |
ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
|
| 46 |
|
| 47 |
# install dependencies
|
| 48 |
+
RUN pip install torch torchvision xformers --index-url https://download.pytorch.org/whl/cu121
|
| 49 |
RUN git submodule update --init --recursive
|
| 50 |
RUN pip install -e ."[tensorrt]"
|
| 51 |
|