teganmosi commited on
Commit
03a728a
1 Parent(s): 59a58ca

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  transformers
2
- torch
3
  datasets
4
  bitsandbytes
5
  accelerate
@@ -21,6 +21,5 @@ tiktoken
21
  kaleido
22
  cohere
23
  typing-extensions>=4.8.0
24
- pydantic>=2.5.1
25
  accelerate
26
  -i https://test.pypi.org/simple/ bitsandbytes
 
1
  transformers
2
+ torch==1.9.1
3
  datasets
4
  bitsandbytes
5
  accelerate
 
21
  kaleido
22
  cohere
23
  typing-extensions>=4.8.0
 
24
  accelerate
25
  -i https://test.pypi.org/simple/ bitsandbytes