in src/huggingface_inference_toolkit/handler.py [0:0]
def __init__( self, model_dir: Union[str, Path], task: Union[str, None] = None, framework: Literal["pt"] = "pt"