(self, model_path: str)
| 695 | return model, tokenizer |
| 696 | |
| 697 | def get_default_conv_template(self, model_path: str) -> Conversation: |
| 698 | if "v0" in remove_parent_directory_name(model_path): |
| 699 | return get_conv_template("one_shot") |
| 700 | return get_conv_template("vicuna_v1.1") |
| 701 | |
| 702 | def raise_warning_for_old_weights(self, model): |
| 703 | if isinstance(model, LlamaForCausalLM) and model.model.vocab_size > 32000: |
nothing calls this directly
no test coverage detected