sileod / tasknet

Easy multi-task learning with HuggingFace Datasets and Trainer
GNU General Public License v3.0
40 stars 3 forks source link

Specify device when loading pipelines #5

Closed tboquet closed 10 months ago

tboquet commented 10 months ago

At the moment we cannot specify a device when loading a pipeline using the load_pipeline helper function. To have a fine grained control over how pipelines are loaded, it would be nice too pass it to the TextClassificationPipeline constructor.