BramVanroy commited on
Commit
ee58a20
·
verified ·
1 Parent(s): c7ad7ab

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -30,12 +30,12 @@ ENV PORT=7860 \
30
  WORKDIR ${HOME}/mateo
31
 
32
  # Clone the repository
33
- RUN git clone --depth 1 --branch v1.4.0 https://github.com/BramVanroy/mateo-demo.git
34
  WORKDIR mateo-demo
35
 
36
  # Install Python dependencies with conditional torch installation
37
  RUN python -m pip install --no-cache-dir --upgrade pip wheel setuptools \
38
- && python -m pip install --no-cache-dir torch==2.2.1+cpu -f https://download.pytorch.org/whl/torch \
39
  && python -m pip install --no-cache-dir --upgrade .
40
 
41
  # Pre-download default models
 
30
  WORKDIR ${HOME}/mateo
31
 
32
  # Clone the repository
33
+ RUN git clone --depth 1 --branch v1.5.0 https://github.com/BramVanroy/mateo-demo.git
34
  WORKDIR mateo-demo
35
 
36
  # Install Python dependencies with conditional torch installation
37
  RUN python -m pip install --no-cache-dir --upgrade pip wheel setuptools \
38
+ && python -m pip install --no-cache-dir torch==2.6.0+cpu -f https://download.pytorch.org/whl/torch \
39
  && python -m pip install --no-cache-dir --upgrade .
40
 
41
  # Pre-download default models