transformers
transformers_stream_generator
optimum
auto-gptq
matplotlib
tiktoken
einops
torchvision