adaface-neurips commited on
Commit
604320a
1 Parent(s): cdf4b99

revert pytorch version

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +2 -2
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM python:3.8-slim
2
  ENV PYTHONUNBUFFERED=1
3
 
4
  RUN RUN apt-get update && \
 
1
+ FROM python:3.10
2
  ENV PYTHONUNBUFFERED=1
3
 
4
  RUN RUN apt-get update && \
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch==1.13.1
2
- torchvision==0.14.1
3
  imageio
4
  imageio-ffmpeg
5
  accelerate
 
1
+ torch
2
+ torchvision
3
  imageio
4
  imageio-ffmpeg
5
  accelerate