Hi, I have a system saving an HF pipeline with the following code: from transformers import pipeline text_generator = pipeline('...') text_generator.save_pretrained('modeldir') How can I re-instantiate that model from a different system What code snippet can do that? I’m looking for something li…

How to load a pipeline saved with pipeline.save_pretrained?

sgugger April 8, 2021, 6:04pm 2

I think pipeline(task, 'modeldir') should work to reload it.

2 Likes