gradio
torch
transformers>=4.36.0
accelerate
psutil
safetensors