(self, model_name, file_path=None, require_model_path=False)
| 427 | |
| 428 | |
| 429 | def fetch_model(self, model_name, file_path=None, require_model_path=False): |
| 430 | fetched_models = [] |
| 431 | fetched_model_paths = [] |
| 432 | for model, model_path, model_name_ in zip(self.model, self.model_path, self.model_name): |
| 433 | if file_path is not None and file_path != model_path: |
| 434 | continue |
| 435 | if model_name == model_name_: |
| 436 | fetched_models.append(model) |
| 437 | fetched_model_paths.append(model_path) |
| 438 | if len(fetched_models) == 0: |
| 439 | print(f"No {model_name} models available.") |
| 440 | return None |
| 441 | if len(fetched_models) == 1: |
| 442 | print(f"Using {model_name} from {fetched_model_paths[0]}.") |
| 443 | else: |
| 444 | print(f"More than one {model_name} models are loaded in model manager: {fetched_model_paths}. Using {model_name} from {fetched_model_paths[0]}.") |
| 445 | if require_model_path: |
| 446 | return fetched_models[0], fetched_model_paths[0] |
| 447 | else: |
| 448 | return fetched_models[0] |
| 449 | |
| 450 | |
| 451 | def to(self, device): |
no outgoing calls
no test coverage detected