(self, model_dir, *args, **kwargs)
| 28 | """ |
| 29 | |
| 30 | def __init__(self, model_dir, *args, **kwargs): |
| 31 | self.model_dir = model_dir |
| 32 | device_name = kwargs.get('device', 'gpu') |
| 33 | verify_device(device_name) |
| 34 | self._device_name = device_name |
| 35 | self.trust_remote_code = kwargs.get( |
| 36 | 'trust_remote_code', |
| 37 | False) or check_model_from_owner_group(model_dir) |
| 38 | |
| 39 | def __call__(self, *args, **kwargs) -> Dict[str, Any]: |
| 40 | return self.postprocess(self.forward(*args, **kwargs)) |
nothing calls this directly
no test coverage detected