(self, model_path: str, from_pretrained_kwargs: dict)
| 1938 | return "codellama" in model_path.lower() |
| 1939 | |
| 1940 | def load_model(self, model_path: str, from_pretrained_kwargs: dict): |
| 1941 | model, tokenizer = super().load_model(model_path, from_pretrained_kwargs) |
| 1942 | model.config.eos_token_id = tokenizer.eos_token_id |
| 1943 | model.config.pad_token_id = tokenizer.pad_token_id |
| 1944 | return model, tokenizer |
| 1945 | |
| 1946 | def get_default_conv_template(self, model_path: str) -> Conversation: |
| 1947 | return get_conv_template("llama-2") |
nothing calls this directly
no test coverage detected