yuntian-deng commited on
Commit
df7bb45
·
1 Parent(s): 9464ffd

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -7
requirements.txt CHANGED
@@ -1,20 +1,20 @@
1
  fastapi
2
  requests
3
  sentencepiece
4
- torch
5
- transformers
6
  uvicorn[standard]
7
  Pillow
8
  huggingface_hub
9
- omegaconf
10
- einops
11
  pytorch-lightning==1.4.2
12
- torchvision
13
- numpy
14
  albumentations
15
  opencv-python
16
  pudb
17
  imageio
18
  imageio-ffmpeg
19
- torch-fidelity
20
  kornia
 
1
  fastapi
2
  requests
3
  sentencepiece
4
+ pytorch=1.7.0
5
+ transformers==4.3.1
6
  uvicorn[standard]
7
  Pillow
8
  huggingface_hub
9
+ omegaconf==2.1.1
10
+ einops==0.3.0
11
  pytorch-lightning==1.4.2
12
+ torchvision=0.8.1
13
+ numpy=1.19.2
14
  albumentations
15
  opencv-python
16
  pudb
17
  imageio
18
  imageio-ffmpeg
19
+ torch-fidelity==0.3.0
20
  kornia