E-Hospital commited on
Commit
907a1e5
·
1 Parent(s): ac44131

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  torch
2
  transformers
 
3
  accelerate
4
  bitsandbytes
5
  fastapi
 
1
  torch
2
  transformers
3
+ transformers[torch]
4
  accelerate
5
  bitsandbytes
6
  fastapi