(self, triton_client: grpcclient.InferenceServerClient)
| 29 | # Controller |
| 30 | class SAMImageEncoder: |
| 31 | def __init__(self, triton_client: grpcclient.InferenceServerClient) -> None: |
| 32 | logger.info("Initialize SAMImageEncoder.") |
| 33 | self.triton_client = triton_client |
| 34 | |
| 35 | @torch.no_grad() |
| 36 | async def run( |
nothing calls this directly
no outgoing calls
no test coverage detected