Sandei commited on
Commit
7eaf453
·
1 Parent(s): ea4df30

numpy error and pytorch errors fix

Browse files
Files changed (3) hide show
  1. Dockerfile +3 -3
  2. memeory.py → memory.py +0 -0
  3. requirements.txt +1 -0
Dockerfile CHANGED
@@ -8,9 +8,9 @@ RUN pip install --upgrade pip
8
 
9
  # Install CPU-only PyTorch
10
  RUN pip install --no-cache-dir \
11
- torch==2.1.2+cpu \
12
- torchvision==0.16.2+cpu \
13
- torchaudio==2.1.2+cpu \
14
  --index-url https://download.pytorch.org/whl/cpu
15
 
16
  RUN pip install --no-cache-dir -r requirements.txt
 
8
 
9
  # Install CPU-only PyTorch
10
  RUN pip install --no-cache-dir \
11
+ torch \
12
+ torchvision \
13
+ torchaudio \
14
  --index-url https://download.pytorch.org/whl/cpu
15
 
16
  RUN pip install --no-cache-dir -r requirements.txt
memeory.py → memory.py RENAMED
File without changes
requirements.txt CHANGED
@@ -10,3 +10,4 @@ accelerate
10
  faiss-cpu
11
  pandas
12
  python-dotenv
 
 
10
  faiss-cpu
11
  pandas
12
  python-dotenv
13
+ numpy<2