(model_dir,
trust_remote_code: bool = True,
use_fast: bool = True,
**kwargs)
| 548 | |
| 549 | @staticmethod |
| 550 | def load_hf_tokenizer(model_dir, |
| 551 | trust_remote_code: bool = True, |
| 552 | use_fast: bool = True, |
| 553 | **kwargs) -> Optional[TransformersTokenizer]: |
| 554 | if (tokenizer := load_hf_tokenizer(model_dir, trust_remote_code, |
| 555 | use_fast, **kwargs)) is not None: |
| 556 | return tokenizer |
| 557 | else: |
| 558 | logger.warning(f"Failed to load tokenizer from {model_dir}") |
| 559 | return None |
| 560 | |
| 561 | @staticmethod |
| 562 | def load_hf_generation_config( |
no test coverage detected