rgp230 commited on
Commit
538de9b
·
1 Parent(s): ba78ba8

fix(retrained_distilbert): Fix for tensor memory leaks

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  altair
2
  pandas
3
  datasets
@@ -9,6 +10,7 @@ langchain_huggingface
9
  langgraph-prebuilt
10
  langchain-tavily
11
  semanticscholar
 
12
  streamlit
13
  transformers==4.55.3
14
  langchain_openai
 
1
+ accelerate==1.9.0
2
  altair
3
  pandas
4
  datasets
 
10
  langgraph-prebuilt
11
  langchain-tavily
12
  semanticscholar
13
+ tavily-python==0.7.10
14
  streamlit
15
  transformers==4.55.3
16
  langchain_openai