Spaces:
Sleeping
Sleeping
leoxing1996
commited on
Commit
·
bd4d838
1
Parent(s):
ca8f50a
update pip install cmd
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
@@ -46,9 +46,10 @@ COPY --chown=user . $HOME/app
|
|
46 |
ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
|
47 |
|
48 |
# install dependencies
|
49 |
-
RUN
|
|
|
50 |
RUN git submodule update --init --recursive
|
51 |
-
RUN
|
52 |
|
53 |
# download models`
|
54 |
RUN mkdir models
|
|
|
46 |
ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
|
47 |
|
48 |
# install dependencies
|
49 |
+
RUN pip3 install --no-cache-dir --upgrade --pre -r /code/requirements.txt
|
50 |
+
RUN pip3 install torch torchvision xformers --index-url https://download.pytorch.org/whl/cu121
|
51 |
RUN git submodule update --init --recursive
|
52 |
+
RUN pip3 install -e ."[tensorrt]"
|
53 |
|
54 |
# download models`
|
55 |
RUN mkdir models
|