torch
transformers[torch]
streamlit
accelerate>=0.21.0