1024m commited on
Commit
de6c693
β€’
1 Parent(s): aca994a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +183 -0
README.md ADDED
@@ -0,0 +1,183 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ base_model:
6
+ - openai/clip-vit-large-patch14-336
7
+ - Qwen/Qwen2-7B
8
+ pipeline_tag: image-text-to-text
9
+ tags:
10
+ - multimodal
11
+ - olmo
12
+ - molmo
13
+ - pixmo
14
+ library_name: transformers
15
+ ---
16
+
17
+ <img src="molmo_logo.png" alt="Logo for the Molmo Project" style="width: auto; height: 50px;">
18
+
19
+ # Molmo 7B-D
20
+
21
+ Molmo is a family of open vision-language models developed by the Allen Institute for AI. Molmo models are trained on PixMo, a dataset of 1 million, highly-curated image-text pairs. It has state-of-the-art performance among multimodal models with a similar size while being fully open-source. You can find all models in the Molmo family [here](https://huggingface.co/collections/allenai/molmo-66f379e6fe3b8ef090a8ca19).
22
+ **Learn more** about the Molmo family [in our announcement blog post](https://molmo.allenai.org/blog).
23
+
24
+ Molmo 7B-D is based on [Qwen2-7B](https://huggingface.co/Qwen/Qwen2-7B) and uses [OpenAI CLIP](https://huggingface.co/openai/clip-vit-large-patch14-336) as vision backbone.
25
+ It performs comfortably between GPT-4V and GPT-4o on both academic benchmarks and human evaluation.
26
+ It powers the **Molmo demo at** [**molmo.allenai.org**](https://molmo.allenai.org).
27
+
28
+ This checkpoint is a **preview** of the Molmo release. All artifacts used in creating Molmo (PixMo dataset, training code, evaluations, intermediate checkpoints) will be made available at a later date, furthering our commitment to open-source AI development and reproducibility.
29
+
30
+ [**Sign up here**](https://docs.google.com/forms/d/e/1FAIpQLSdML1MhNNBDsCHpgWG65Oydg2SjZzVasyqlP08nBrWjZp_c7A/viewform) to be the first to know when artifacts are released.
31
+
32
+ Quick links:
33
+ - πŸ’¬ [Demo](https://molmo.allenai.org/)
34
+ - πŸ“‚ [All Models](https://huggingface.co/collections/allenai/molmo-66f379e6fe3b8ef090a8ca19)
35
+ - πŸ“ƒ [Paper](https://molmo.allenai.org/paper.pdf)
36
+ - πŸŽ₯ [Blog with Videos](https://molmo.allenai.org/blog)
37
+
38
+
39
+ ## Quick Start
40
+
41
+ To run Molmo, first install dependencies:
42
+
43
+ ```bash
44
+ pip install einops torchvision
45
+ ```
46
+
47
+ Then, follow these steps:
48
+
49
+ ```python
50
+ from transformers import AutoModelForCausalLM, AutoProcessor, GenerationConfig
51
+ from PIL import Image
52
+ import requests
53
+
54
+ # load the processor
55
+ processor = AutoProcessor.from_pretrained(
56
+ 'allenai/Molmo-7B-D-0924',
57
+ trust_remote_code=True,
58
+ torch_dtype='auto',
59
+ device_map='auto'
60
+ )
61
+
62
+ # load the model
63
+ model = AutoModelForCausalLM.from_pretrained(
64
+ 'allenai/Molmo-7B-D-0924',
65
+ trust_remote_code=True,
66
+ torch_dtype='auto',
67
+ device_map='auto'
68
+ )
69
+
70
+ # process the image and text
71
+ inputs = processor.process(
72
+ images=[Image.open(requests.get("https://picsum.photos/id/237/536/354", stream=True).raw)],
73
+ text="Describe this image."
74
+ )
75
+
76
+ # move inputs to the correct device and make a batch of size 1
77
+ inputs = {k: v.to(model.device).unsqueeze(0) for k, v in inputs.items()}
78
+
79
+ # generate output; maximum 200 new tokens; stop generation when <|endoftext|> is generated
80
+ output = model.generate_from_batch(
81
+ inputs,
82
+ GenerationConfig(max_new_tokens=200, stop_strings="<|endoftext|>"),
83
+ tokenizer=processor.tokenizer
84
+ )
85
+
86
+ # only get generated tokens; decode them to text
87
+ generated_tokens = output[0,inputs['input_ids'].size(1):]
88
+ generated_text = processor.tokenizer.decode(generated_tokens, skip_special_tokens=True)
89
+
90
+ # print the generated text
91
+ print(generated_text)
92
+
93
+ # >>> This image features an adorable black Labrador puppy, captured from a top-down
94
+ # perspective. The puppy is sitting on a wooden deck, which is composed ...
95
+ ```
96
+
97
+ ## Evaluations
98
+
99
+ | Model | Average Score on 11 Academic Benchmarks | Human Preference Elo Rating |
100
+ |-----------------------------|-----------------------------------------|-----------------------------|
101
+ | Molmo 72B | 81.2 | 1077 |
102
+ | **Molmo 7B-D (this model)** | **77.3** | **1056** |
103
+ | Molmo 7B-O | 74.6 | 1051 |
104
+ | MolmoE 1B | 68.6 | 1032 |
105
+ | GPT-4o | 78.5 | 1079 |
106
+ | GPT-4V | 71.1 | 1041 |
107
+ | Gemini 1.5 Pro | 78.3 | 1074 |
108
+ | Gemini 1.5 Flash | 75.1 | 1054 |
109
+ | Claude 3.5 Sonnet | 76.7 | 1069 |
110
+ | Claude 3 Opus | 66.4 | 971 |
111
+ | Claude 3 Haiku | 65.3 | 999 |
112
+ | Qwen VL2 72B | 79.4 | 1037 |
113
+ | Qwen VL2 7B | 73.7 | 1025 |
114
+ | Intern VL2 LLAMA 76B | 77.1 | 1018 |
115
+ | Intern VL2 8B | 69.4 | 953 |
116
+ | Pixtral 12B | 69.5 | 1016 |
117
+ | Phi3.5-Vision 4B | 59.7 | 982 |
118
+ | PaliGemma 3B | 50.0 | 937 |
119
+ | LLAVA OneVision 72B | 76.6 | 1051 |
120
+ | LLAVA OneVision 7B | 72.0 | 1024 |
121
+ | Cambrian-1 34B | 66.8 | 953 |
122
+ | Cambrian-1 8B | 63.4 | 952 |
123
+ | xGen - MM - Interleave 4B | 59.5 | 979 |
124
+ | LLAVA-1.5 13B | 43.9 | 960 |
125
+ | LLAVA-1.5 7B | 40.7 | 951 |
126
+
127
+ *Benchmarks: AI2D test, ChartQA test, VQA v2.0 test, DocQA test, InfographicVQA test, TextVQA val, RealWorldQA, MMMU val, MathVista testmini, CountBenchQA, Flickr Count (we collected this new dataset that is significantly harder than CountBenchQA).*
128
+
129
+ ## FAQs
130
+
131
+ ### I'm getting an error a broadcast error when processing images!
132
+
133
+ Your image might not be in RGB format. You can convert it using the following code snippet:
134
+
135
+ ```python
136
+ from PIL import Image
137
+
138
+ image = Image.open(...)
139
+
140
+ if image.mode != "RGB":
141
+ image = image.convert("RGB")
142
+ ```
143
+
144
+ ### Molmo doesn't work great with transparent images!
145
+
146
+ We received reports that Molmo models might struggle with transparent images.
147
+ For the time being, we recommend adding a white or dark background to your images before passing them to the model. The code snippet below shows how to do this using the Python Imaging Library (PIL):
148
+
149
+ ```python
150
+
151
+ # Load the image
152
+ url = "..."
153
+ image = Image.open(requests.get(url, stream=True).raw)
154
+
155
+ # Convert the image to grayscale to calculate brightness
156
+ gray_image = image.convert('L') # Convert to grayscale
157
+
158
+ # Calculate the average brightness
159
+ stat = ImageStat.Stat(gray_image)
160
+ average_brightness = stat.mean[0] # Get the average value
161
+
162
+ # Define background color based on brightness (threshold can be adjusted)
163
+ bg_color = (0, 0, 0) if average_brightness > 127 else (255, 255, 255)
164
+
165
+ # Create a new image with the same size as the original, filled with the background color
166
+ new_image = Image.new('RGB', image.size, bg_color)
167
+
168
+ # Paste the original image on top of the background (use image as a mask if needed)
169
+ new_image.paste(image, (0, 0), image if image.mode == 'RGBA' else None)
170
+
171
+ # Now you can pass the new_image to Molmo
172
+ processor = AutoProcessor.from_pretrained(
173
+ 'allenai/Molmo-7B-D-0924',
174
+ trust_remote_code=True,
175
+ torch_dtype='auto',
176
+ device_map='auto'
177
+ )
178
+ ```
179
+
180
+ ## License and Use
181
+
182
+ This model is licensed under Apache 2.0. It is intended for research and educational use.
183
+ For more information, please see our [Responsible Use Guidelines](https://allenai.org/responsible-use).