pytorch/serve

Having two different endpoints for a single model

Open

#2 058 ouverte le 5 janv. 2023

Voir sur GitHub
 (2 commentaires) (0 réactions) (0 assignés)Java (790 forks)batch import
enhancementhelp wantedtriaged

Métriques du dépôt

Stars
 (3 844 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

Hello!

Is it possible to create two different endpoints for a single model? So it can work in two separate modes and batch accordingly depending on the input (e.g., it can generate an image or perform upscaling given an input). I can make two different models but then the memory will be double, while actually no extra memory is needed (as the model weights are the same). Ideally, I want to have the same model weights but two different endpoints with different batching settings.

Best, Valentin

Guide contributeur