radames HF staff commited on
Commit
0ece197
1 Parent(s): 20c216c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -6,6 +6,7 @@ COPY ./requirements.txt /code/requirements.txt
6
  COPY ./pre-requirements.txt /code/pre-requirements.txt
7
 
8
  RUN pip install --no-cache-dir --upgrade -r /code/pre-requirements.txt
 
9
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
10
 
11
  # Set up a new user named "user" with user ID 1000
 
6
  COPY ./pre-requirements.txt /code/pre-requirements.txt
7
 
8
  RUN pip install --no-cache-dir --upgrade -r /code/pre-requirements.txt
9
+ RUN pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu117
10
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
11
 
12
  # Set up a new user named "user" with user ID 1000