torch
transformers
tokenizers
PyYAML