pytorch/serve

Having two different endpoints for a single model

Open

#2.058 aberto em 5 de jan. de 2023

Ver no GitHub
 (2 comments) (0 reactions) (0 assignees)Java (790 forks)batch import
enhancementhelp wantedtriaged

Métricas do repositório

Stars
 (3.844 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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

Guia do colaborador