Chaitanya Sagar Gurujula commited on
Commit
2897777
·
1 Parent(s): 2799123

smollm2 text gen code dependencies

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,6 +1,8 @@
1
  fastapi==0.68.0
2
  uvicorn==0.15.0
3
  jinja2==3.0.1
 
 
4
  torch
5
  torchtune
6
  transformers
 
1
  fastapi==0.68.0
2
  uvicorn==0.15.0
3
  jinja2==3.0.1
4
+ torchao
5
+ triton
6
  torch
7
  torchtune
8
  transformers