INNER CODE UNIT · Python

__init__

onnxsim/onnxsim · onnxsim/rpc/server.py:46

    def __init__(
        self,
        model_bytes: bytes,
        providers: Optional[Sequence[str]],
        single_threaded: bool,
    ):
        self.model_bytes = model_bytes
        self.providers = list(providers) if providers else None
        self.single_threaded = single_threaded
        self.session = None
        try:
            import onnxruntime as ort
        except ImportError:
            return
        available = ort.get_available_providers()
        chosen = self.providers or ["CPUExecutionProvider"]
        unknown = [p for p in chosen if p not in available]
        if unknown:

View source record →