Valentin Peron commited on
Commit
b90d46c
·
1 Parent(s): 5e0990b

build: Update requirements.txt with torch version constraint

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. requirements.txt +4 -2
Dockerfile CHANGED
@@ -4,6 +4,7 @@ RUN useradd -m -u 1000 user
4
  WORKDIR /app
5
 
6
  COPY --chown=user ./requirements.txt requirements.txt
 
7
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
8
 
9
  COPY --chown=user . /app
 
4
  WORKDIR /app
5
 
6
  COPY --chown=user ./requirements.txt requirements.txt
7
+
8
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
9
 
10
  COPY --chown=user . /app
requirements.txt CHANGED
@@ -1,8 +1,10 @@
1
  fastapi==0.74.*
2
  requests==2.27.*
3
  sentencepiece==0.1.*
4
- torch==1.11.*
5
  transformers==4.*
6
  uvicorn[standard]==0.17.*
7
  numpy<2.0
8
- facenet-pytorch==2.6.0
 
 
 
1
  fastapi==0.74.*
2
  requests==2.27.*
3
  sentencepiece==0.1.*
4
+ torch>=2.2.0,<2.3.0
5
  transformers==4.*
6
  uvicorn[standard]==0.17.*
7
  numpy<2.0
8
+ facenet-pytorch==2.6.0
9
+ starlette==0.17.1
10
+ pydantic<2.0.0,>=1.6.2