torch
numpy
gradio
transformers
einops
peft
torchvision