(self, model_path: str, from_pretrained_kwargs: dict)
| 1513 | return "llama-2" in model_path.lower() |
| 1514 | |
| 1515 | def load_model(self, model_path: str, from_pretrained_kwargs: dict): |
| 1516 | model, tokenizer = super().load_model(model_path, from_pretrained_kwargs) |
| 1517 | model.config.eos_token_id = tokenizer.eos_token_id |
| 1518 | model.config.pad_token_id = tokenizer.pad_token_id |
| 1519 | return model, tokenizer |
| 1520 | |
| 1521 | def get_default_conv_template(self, model_path: str) -> Conversation: |
| 1522 | return get_conv_template("llama-2") |
nothing calls this directly
no test coverage detected