RishiRP commited on
Commit
a0bb3da
·
verified ·
1 Parent(s): 0c070e3

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -0
requirements.txt CHANGED
@@ -8,3 +8,8 @@ websockets>=10,<13
8
  # transformers==4.44.2
9
  # accelerate>=0.34.0
10
  # bitsandbytes==0.43.1
 
 
 
 
 
 
8
  # transformers==4.44.2
9
  # accelerate>=0.34.0
10
  # bitsandbytes==0.43.1
11
+
12
+ # Add PyTorch
13
+ torch==2.2.2
14
+
15
+ transformers==4.44.2