🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
GPU text generation: mMoved the encoded_prompt to correct device
A
alberduris committed
81d6841b4be25a164235975e5ebdcf99d7a26633
Parent: dd4df80
Committed by Lysandre Debut <lysandre@huggingface.co>
on 1/6/2020, 2:11:12 PM