tejoess commited on
Commit
292b89e
·
1 Parent(s): da78d4a

Add custom handler for Inference Endpoint deployment

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -6
requirements.txt CHANGED
@@ -1,6 +1,5 @@
1
- # requirements.txt
2
- transformers
3
- torch
4
- accelerate
5
- timm
6
- einops
 
1
+ transformers==4.35.2
2
+ torch==2.1.0
3
+ accelerate==0.24.1
4
+ timm==0.9.8
5
+ einops==0.7.0