gaidasalsaa commited on
Commit
806ab2d
·
1 Parent(s): 46465cb

Fix: Locked Transformers version to 4.35.2 for PyTorch 2.1 compatibility

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +0 -1
Dockerfile CHANGED
@@ -3,7 +3,7 @@ FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
3
  WORKDIR /app
4
 
5
  COPY requirements.txt /app/requirements.txt
6
- RUN pip install --no-cache-dir fastapi uvicorn[standard] pydantic transformers tweepy numpy
7
 
8
  COPY . /app
9
 
 
3
  WORKDIR /app
4
 
5
  COPY requirements.txt /app/requirements.txt
6
+ RUN pip install --no-cache-dir fastapi uvicorn[standard] pydantic "transformers==4.35.2" tweepy numpy
7
 
8
  COPY . /app
9
 
requirements.txt CHANGED
@@ -1,7 +1,6 @@
1
  fastapi
2
  uvicorn[standard]
3
  python-multipart
4
- transformers
5
  pandas
6
  tweepy
7
  emoji
 
1
  fastapi
2
  uvicorn[standard]
3
  python-multipart
 
4
  pandas
5
  tweepy
6
  emoji