acecalisto3 commited on
Commit
49da029
·
verified ·
1 Parent(s): 0f5752d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -2,10 +2,12 @@ transformers
2
  transformers[torch]
3
  huggingface_hub
4
  torch
 
 
5
  tf-keras
6
  tensor
7
  tensorflow
8
- tensorrt-cu12
9
  flax
10
  accelerate==0.21.0
11
  autopep8
 
2
  transformers[torch]
3
  huggingface_hub
4
  torch
5
+ nvidia-tensorrt
6
+ nvidia-pyindex
7
  tf-keras
8
  tensor
9
  tensorflow
10
+ #"tensorrt-cu12
11
  flax
12
  accelerate==0.21.0
13
  autopep8