File size: 290 Bytes
e58ac13 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import torch
import numpy as np
from pytriton.model_config import ModelConfig, Tensor
from pytriton.triton import Triton
# Connecting inference callback with Triton Inference Server
with Triton() as triton:
# Load model into Triton Inference Server
triton
triton.serve()
|