(
self,
workspace: NNFolderWorkspace,
metadata: NetworkMetadata,
)
| 329 | ) |
| 330 | |
| 331 | def _download_models( |
| 332 | self, |
| 333 | workspace: NNFolderWorkspace, |
| 334 | metadata: NetworkMetadata, |
| 335 | ) -> Tuple[NetworkModel]: |
| 336 | # No fpath provided for onnx files, download them from HuggingFace repo. |
| 337 | return self.frameworks_cmd.generate_and_download_framework( |
| 338 | metadata, workspace |
| 339 | ).onnx |
| 340 | |
| 341 | @abstractmethod |
| 342 | def run_trt( |
no test coverage detected