(self, model_path: str, from_pretrained_kwargs: dict)
| 956 | return "pythia" in model_path.lower() |
| 957 | |
| 958 | def load_model(self, model_path: str, from_pretrained_kwargs: dict): |
| 959 | model, tokenizer = super().load_model(model_path, from_pretrained_kwargs) |
| 960 | model.config.eos_token_id = tokenizer.eos_token_id |
| 961 | model.config.pad_token_id = tokenizer.pad_token_id |
| 962 | return model, tokenizer |
| 963 | |
| 964 | |
| 965 | class StableLMAdapter(BaseModelAdapter): |
nothing calls this directly
no test coverage detected