:py:mod:`wsinfer.modellib.models`
=================================

.. py:module:: wsinfer.modellib.models


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   wsinfer.modellib.models.LocalModelTorchScript



Functions
~~~~~~~~~

.. autoapisummary::

   wsinfer.modellib.models.get_registered_model
   wsinfer.modellib.models.get_pretrained_torch_module
   wsinfer.modellib.models.jit_compile



.. py:class:: LocalModelTorchScript


   Bases: :py:obj:`wsinfer_zoo.client.Model`


.. py:function:: get_registered_model(name: str) -> wsinfer_zoo.client.HFModelTorchScript


.. py:function:: get_pretrained_torch_module(model: wsinfer_zoo.client.HFModelTorchScript | LocalModelTorchScript) -> torch.nn.Module

   Get a PyTorch Module with weights loaded.


.. py:function:: jit_compile(model: torch.nn.Module) -> torch.jit.ScriptModule | torch.nn.Module | Callable

   JIT-compile a model for inference.

   A torchscript model may be JIT compiled here as well.