aarticerebras commited on
Commit
42de6b3
·
verified ·
1 Parent(s): 05081db

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -0
README.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ {}
3
+ ---
4
+ # Model Card for cerebras/Cerebras-LLaVA-13B
5
+ The checkpoints consists of Language encoder and projector weights of multimodal LLaVA-13B model trained with our Cerebras implementation and training recipe.
6
+ The vision encoder checkpoints for this model can be found at [cerebras/Cerebras-ViT-L-336-patch14-llava7b-ShareGPT4V](https://huggingface.co/cerebras/Cerebras-ViT-L-336-patch14-llava7b-ShareGPT4V)
7
+
8
+ **Note**: _ShareGPT4V_ is added to the vision model name to ensure correct loading of checkpoints in [LLaVA source repo](https://github.com/haotian-liu/LLaVA/blob/main/llava/model/multimodal_encoder/builder.py#L8)
9
+
10
+ For full details of this model and training details, please read our upcoming blog post.
11
+
12
+ ## License
13
+ Cerebras-Llava is licensed under the LLAMA 2 Community License, Copyright (c) Meta Platforms, Inc. All Rights Reserved.
14
+
15
+ ## Model Architecture
16
+ Cerebras-LLaVA-13B is a transformer model with the following architecture details
17
+ * Vision encoder: [CLIP-VisionModel-Large](cerebras/Cerebras-ViT-L-336-patch14-llava7b-ShareGPT4V). It handles images of size 336 x 336 with patch size of 14
18
+ * Large Language Model: Pretrained from Vicuna-13B checkpoints and instruction finetuned on various datasets.
19
+ * Projector: the projector module that connects the LLM and Vision encoder part consists of two linear layers with gelu activation (mlp2x-gelu)
20
+
21
+ ## Loading the model
22
+ This model can directly be loaded using the [LLaVa source code repository](https://github.com/haotian-liu/LLaVA). For installation, please refer to the [instructions in source code repository](https://github.com/haotian-liu/LLaVA?tab=readme-ov-file#install).
23
+ We perform all our evaluations using the LLaVA source code repository scripts.
24
+
25
+ ```
26
+ from llava.model.builder import load_pretrained_model
27
+ from llava.mm_utils import get_model_name_from_path
28
+ from llava.eval.run_llava import eval_model
29
+
30
+ model_path = "cerebras/Cerebras-LLaVA-13B"
31
+
32
+ tokenizer, model, image_processor, context_len = load_pretrained_model(
33
+ model_path=model_path,
34
+ model_base=None,
35
+ model_name=get_model_name_from_path(model_path)
36
+ )
37
+ ```
38
+
39
+ ## Intended Use
40
+ _Primary intended uses_: The primary use of LLaVA is research on large multimodal models and chatbots.
41
+
42
+ _Primary intended users_: The primary intended users of the model are researchers(both academic and industry) in computer vision, natural language processing, machine learning, and artificial intelligence
43
+
44
+ ## Limitations and Bias
45
+ The pre-training dataset may have contained offensive or inappropriate content, even after applying data cleansing filters, which can be reflected in the model-generated text.
46
+ We recommend that users exercise caution when using these models for their applications or any use case that may cause deliberate or unintentional harm to others. This model is for demonstration purpose only.
47
+ ## Acknowledgements
48
+ We are thankful to all Cerebras engineers that made this work possible.