Skip to content

PyTriton 0.4.0

Compare
Choose a tag to compare
@nv-blazejkubiak nv-blazejkubiak released this 24 Oct 19:40
· 129 commits to main since this release
  • New: Remote Mode - PyTriton can be used to connect to a remote Triton Inference Server

    • Introduced RemoteTriton class which can be used to connect to a remote Triton Inference Server running on the same machine, by passing triton url.
    • Changed Triton lifecycle - now the Triton Inference Server is started while entering the context. This allows to load models dynamically to the running server while calling the bind method. It is still allowed to create Triton instance without entering the context and bind models before starting the server (in this case the models are lazy loaded when calling run or serve method like it worked before).
    • In RemoteTriton class, calling enter or connect method connects to triton server, so we can safely load models while binding inference functions (if RemoteTriton is used without context manager, models are lazy loaded when calling connect or serve method).
  • Change: "batch" decorator raises a ValueError if any of the outputs have a different batch size than expected.

  • Fix: gevent resources leak in FuturesModelClient

  • Version of Triton Inference Server embedded in wheel: 2.36.0