Skip to content

Instantly share code, notes, and snippets.

@ShopifyEng
Created March 10, 2023 14:45
Unlocking Real-time Predictions with Shopify's Machine Learning Platform: Settings example for loading a model using the HuggingFace Inference Runtime
{
"name": "transformer",
"implementation": "mlserver_huggingface.HuggingFaceRuntime",
"parallel_workers": 0,
"parameters": {
"extra": {
"task": "text-generation",
"pretrained_model": "distilgpt2"
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment