sudokara commited on
Commit
6faf4b3
·
1 Parent(s): 92d5951

added accelerate to requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -16,4 +16,5 @@ transformers==4.46.2
16
  typing_extensions==4.12.2
17
  urllib3==2.2.3
18
  --find-links https://download.pytorch.org/whl/torch_stable.html
19
- torch
 
 
16
  typing_extensions==4.12.2
17
  urllib3==2.2.3
18
  --find-links https://download.pytorch.org/whl/torch_stable.html
19
+ torch
20
+ accelerate>=0.26.0