jordigonzm commited on
Commit
acfb781
1 Parent(s): 50c763f

update requerimets

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -0
requirements.txt CHANGED
@@ -7,3 +7,6 @@ spaces==0.28.3
7
  torch==2.2.0
8
  transformers==4.41.0
9
  numpy<2.0.0
 
 
 
 
7
  torch==2.2.0
8
  transformers==4.41.0
9
  numpy<2.0.0
10
+ pycuda
11
+ pynvml
12
+ tensorflow