handle load_model splat

2023-10-23 21:55:05 -04:00
parent ab9d12ce34
commit b885169229
2 changed files with 2 additions and 1 deletions
--- a/examples/multimodal/llava-mistral.yml
+++ b/examples/multimodal/llava-mistral.yml
@@ -2,6 +2,7 @@ base_model: mistralai/Mistral-7B-v0.1
 model_type: MistralForCausalLM
 tokenizer_type: LlamaTokenizer
 is_mistral_derived_model: true
+multimodal: true

 vision_tower: openai/clip-vit-large-patch14
 tune_mm_mlp_adapter: true