torch>=1.9
pytorch-lightning<1.6.0,>=1.5.0
torchmetrics<0.8.0,>=0.5.0
transformers<4.18,>=4.1
hydra-core==1.1.1
argcomplete==1.12.3
wandb>=0.12.0
datasets<1.19,>=1.12
nltk<4.0.0,>=3.4.5
seqeval==1.2.*
rouge-score==0.0.4
sacrebleu==2.0.0
rich>=11.0.0
sentencepiece==0.1.96

[all]
fastapi==0.68.1
uvicorn[standard]==0.15.0
streamlit<1.5.0,>=1.4.0
pdoc3==0.10.0
st-annotated-text==2.0.0
plotly==5.5.0

[demo]
st-annotated-text==2.0.0
streamlit<1.5.0,>=1.4.0

[describe]
st-annotated-text==2.0.0
plotly==5.5.0
streamlit<1.5.0,>=1.4.0

[docs]
pdoc3==0.10.0

[fastapi]
fastapi==0.68.1

[pdoc3]
pdoc3==0.10.0

[plotly]
plotly==5.5.0

[serve]
fastapi==0.68.1
uvicorn[standard]==0.15.0

[st-annotated-text]
st-annotated-text==2.0.0

[streamlit]
streamlit<1.5.0,>=1.4.0

[uvicorn[standard]]
uvicorn[standard]==0.15.0
