transformers

ca9b6f32 - specify tensor device in beam search