(self, model_name,api_key=None, trust_remote_code=True, caller=None)
| 12278 | self.caller = caller |
| 12279 | |
| 12280 | def loader(self, model_name,api_key=None, trust_remote_code=True, caller=None): |
| 12281 | |
| 12282 | self.model_name = model_name |
| 12283 | self.api_key= api_key |
| 12284 | self.trust_remote_code=trust_remote_code |
| 12285 | self.caller = caller |
| 12286 | |
| 12287 | if self.caller == "generative_model": |
| 12288 | return self.load_generative_model() |
| 12289 | |
| 12290 | if self.caller == "embedding_model": |
| 12291 | return self.load_embedding_model() |
| 12292 | |
| 12293 | if self.caller == "tokenizer": |
| 12294 | return self.load_tokenizer() |
| 12295 | |
| 12296 | def load_generative_model(self): |
| 12297 |
no test coverage detected