flask == 2.2.2 gunicorn == 21.2.0 pytest torch == 2.0.1 transformers == 4.31.0 # for the tokenizer of the baseline model protobuf == 4.24.0 sentencepiece==0.1.99