vitorcalvi commited on
Commit
54939b3
Β·
1 Parent(s): a378df1
Files changed (1) hide show
  1. requirements.txt +18 -18
requirements.txt CHANGED
@@ -1,7 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # CUDA-enabled PyTorch packages
2
- torch==2.0.1+cu118
3
- torchvision==0.15.2+cu118
4
- torchaudio==2.0.2+cu118
5
  -f https://download.pytorch.org/whl/torch_stable.html
6
 
7
  # Core dependencies
@@ -49,18 +63,4 @@ uvicorn==0.30.1
49
  # Any other necessary dependencies
50
  # Add your additional dependencies here
51
 
52
- sentencepiece
53
- tiktoken
54
- jsonlines
55
- diffusers==0.30.1
56
- accelerate==0.30.0
57
- torchvision
58
- numpy==1.26.4
59
- imageio
60
- imageio-ffmpeg
61
- timm
62
- transformers
63
- opencv-python-headless
64
- einops
65
- tensorboardX
66
- ipython
 
1
+ sentencepiece
2
+ tiktoken
3
+ jsonlines
4
+ diffusers==0.30.1
5
+ accelerate==0.30.0
6
+ torchvision
7
+ numpy==1.26.4
8
+ imageio
9
+ imageio-ffmpeg
10
+ timm
11
+ transformers
12
+ opencv-python-headless
13
+ einops
14
+ tensorboardX
15
+ ipython
16
+
17
  # CUDA-enabled PyTorch packages
18
+ torchaudio
 
 
19
  -f https://download.pytorch.org/whl/torch_stable.html
20
 
21
  # Core dependencies
 
63
  # Any other necessary dependencies
64
  # Add your additional dependencies here
65
 
66
+