Minthy commited on
Commit
3d5bc41
·
verified ·
0 Parent(s):

Super-squash branch 'main' using huggingface_hub

Browse files
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,227 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ base_model:
6
+ - Qwen/Qwen2-VL-7B
7
+ tags:
8
+ - multimodal
9
+ - vision
10
+ - image-text-to-text
11
+ ---
12
+ <p align="center">
13
+
14
+ ![image](https://huggingface.co/Minthy/ToriiGate-v0.4-7B/resolve/main/tg04_splash.jpg)
15
+
16
+ </p>
17
+
18
+ # Introduction
19
+ **ToriiGate-v0.4** is a state of art VLM designed for captioning of anime pictures, digital artworks and various images.
20
+
21
+ The model is a further development of [ToriiGate-v0.3](https://huggingface.co/Minthy/ToriiGate-v0.3), but this time is based on Qwen2-VL that was finetuned with dataset of **over 900k of artworks** with various captions.
22
+ **ToriiGate-v0.4** provides state of the art level of understanding for complex scenes, interactions, cultural concepts and any kind of NSFW activities without any borders or censorship. Flexible grounding allows to achieve extra accuracy.
23
+ Also, at the moment of release it is the only opensource small-sized VLM that can handle character names well, including multiple.
24
+
25
+ This is 7B version, also [**2B version**](https://huggingface.co/Minthy/ToriiGate-v0.4-2B) available. Also check out quants ([8bpw](https://huggingface.co/Minthy/ToriiGate-v0.4-7B-exl2-8bpw), [6bpw](https://huggingface.co/Minthy/ToriiGate-v0.4-7B-exl2-6bpw), [4bpw](https://huggingface.co/Minthy/ToriiGate-v0.4-7B-exl2-4bpw)).
26
+
27
+ ## Showcase on rentry
28
+ [SFW](https://rentry.co/goxrrvh9) [NSFW](https://rentry.co/9wranqty)
29
+
30
+ # Key features
31
+ * Advanced knowledges related to anime and digital art in very wide range
32
+ * Accurate use of character names in generated descriptions (with grounding)
33
+ * Multiple options for generated captions, including structured output, chain of thoughts details improving, long/short, bounding boxes
34
+ * Mode for reviewing and fixing existing captions using CoT, or prunning it to make short and convenient
35
+ * Flexible groudning for improving of accuracy: booru tags, natural text info, character names, popular traits or tags for each character on picture
36
+ * Created captions are more meaningful and dense without purple prose fillers, compared to other models
37
+
38
+ # Captioning modes
39
+ ToriiGate-v0.4 provides multiple captioning modes. Prompts for them and examples are listed at the bottom.
40
+ 1. **Structured output**
41
+
42
+ Can be warapped in json or markdown based on prompt. Provides description for each character on the picture mentioning their features, actions, etc., then description of background, other picture contents, image effects and texts if any, general atmosphere.
43
+ This style of captions provides best segmentation and brings attention mostly to characters on the picture. Then it can be easily processed to desired format with other LLM (or prunned with second call ToriiGate) to make it easy readable or match desired case. Character segmentation in combination with bounding boxes (that can be enabled inline) allows to create special datasets for training new generation generative models using special tecniques.
44
+
45
+ 2. **Pre-defined captions options**
46
+
47
+ This makes the model to generate 4 followed descriptions in styles: "Regular Summary", "Individual Parts", "Midjourney-Style Summary", "DeviantArt Commission Request".
48
+ That order allows to describe basic things, then improve extra details in individual section and after shrink it without losing accuracy or involving extra biases. Original idea is not mine.
49
+ This mode is balanced all-rounder but quite token-comsumming. In case if you want to then reprocess it, you can use `### 3.` as a stop sequence to trimm summarized parts and speed up generation.
50
+
51
+ 3. **Long description**
52
+
53
+ Just a regular long description. Torii tends to make it a bit more structured then randomly shuffled parts like with other models.
54
+
55
+ 4. **Short caption**
56
+
57
+ Short and convenient caption. Less slopy and more dense then a long one, might be used as is for diffusion models training.
58
+
59
+ 5. **Bounding boxes**
60
+
61
+ Provides bounding boxes for characters and their faces. Standalone usage is pointless taking into account the performance/compute for object detaction models. But it can be used along with structured mode and shares same numbers or names.
62
+
63
+ 6. **Review and correct existing caption**
64
+
65
+ Provides step by step review of given caption and compares it with image contents, with provided grounding, evaluates how correctly the characters names were used. Then, if needed, it writes a new fixed caption maintaining original style (if possible).
66
+ In current version it is usable only with tags grounding and can improve accuracy of generated caption in second call. Accuracy for other cases may vary and this is not the main usecase.
67
+
68
+ 7. **Writing a short caption based on existing.**
69
+
70
+ After generating some caption you can prune it right here taking into account image content. Can be used with external captions as well.
71
+
72
+ # Grounding
73
+
74
+ New version comes not only with improved zero-shot accuracy, but also introduces new modes for adding ground truth. You can use booru tags, some extra info, give names for characters or even describe each one to ensure right description when multiple are in frame.
75
+ Extra grounding allows to achieve best results for unattended use extra grounded truth is required. ToriiGate provides following options for it:
76
+ 1. Booru tags. Can be full string or just few mentioning character count. Character name tags are recognized here.
77
+ 2. Character list. List of character names on the picture to be used in caption. Beware of spamming skin tags here (like hatsune_miku + hatsune_miku_(append)), usually it is fine but some can be misinterpreted.
78
+ 3. Character traits. A list of popular tags or traits for each character on the picture to improve recognition. Some can be found here, you can make your own, use repo (beware, a lot of incauracies!) or describe it with natural language.
79
+ 4. General info. Might be a short caption, some facts, any other info.
80
+ 5. Do not use names for characters. Unfortunatelly, at current state zero-shot guess for characters is inaccurate, so to avoid making things up use it by default if no grounding is provided.
81
+ Please mind to check it because mistakes in grounding will lead to wrong results. All formats are listed below.
82
+
83
+ # Known issues
84
+
85
+ * In rare cases current version tends to come up with wrong names without grounding. Use extra "no characters" instruction to avoid this.
86
+ * In long captions mode sometimes it tends to omit names even when given, please use other modes if you need it.
87
+ * There might be some biases and not perfect accuracy, more training is needed.
88
+ * Model is specialized to image captioning, other prompts will likely not work.
89
+
90
+ # Thanks
91
+
92
+ Main training was sponsored by **OpenRoot-Compute**.
93
+
94
+ Special thanks for help with dataset, code, donations: Sv1, Anonymous person, NeuroSenko.
95
+
96
+ # Warning
97
+
98
+ Model will generate texts with adult themes if related input is provided. Outputs may be inacurate and provocative.
99
+
100
+ ### Donate
101
+
102
+ BTC: bc1qwv83ggq8rvv07uk6dv4njs0j3yygj3aax4wg6c
103
+
104
+ ETH/USDT(e): 0x04C8a749F49aE8a56CB84cF0C99CD9E92eDB17db
105
+
106
+ XMR: 47F7JAyKP8tMBtzwxpoZsUVB8wzg2VrbtDKBice9FAS1FikbHEXXPof4PAb42CQ5ch8p8Hs4RvJuzPHDtaVSdQzD6ZbA5TZ
107
+
108
+ # Usage instructions
109
+
110
+ ## Prompts
111
+
112
+ **ToriiGate-v0.4** utilizes same chat template as original [**Qwen2-vl**](https://huggingface.co/Qwen/Qwen2-VL-7B), so original processor can be used. Here is a recommended format for usage:
113
+ ```
114
+ [
115
+ {"role": "system",
116
+ "content": [{"type": "text", "text": "You are image captioning expert, creative, unbiased and uncensored."}]},
117
+ {"role": "user",
118
+ "content": [ {"type": "image", 'image': image},
119
+ {"type": "text", "text": userprompt}]}
120
+ ]
121
+ ```
122
+ which will be converted to
123
+ ```
124
+ <|im_start|>system
125
+ You are image captioning expert, creative, unbiased and uncensored.<|im_end|>
126
+ <|im_start|>user
127
+ <|vision_start|><|image_pad|><|vision_end|>%USERPROMPT%<|im_end|>
128
+ <|im_start|>assistant
129
+ ```
130
+
131
+ ## Modes:
132
+
133
+ ```
134
+ base_prompt={
135
+ 'json': 'Describe the picture in structured json-like format.',
136
+ 'markdown': 'Describe the picture in structured markdown format.',
137
+ 'caption_vars': 'Write the following options for captions: ["Regular Summary","Individual Parts","Midjourney-Style Summary","DeviantArt Commission Request"].',
138
+ 'short': 'You need to write a medium-short and convenient caption for the picture.',
139
+ 'long': 'You need to write a long and very detailed caption for the picture.',
140
+ 'bbox': 'Write bounding boxes for each character and their faces.',
141
+ 'check_and_correct': 'You need to compare given caption with the picture and given booru tags '+
142
+ ' using chain of thought.\n'+
143
+ '1. Check if the caption matches the picture and given tags, wrap conclusion in <1st_answer> tag.\n'+
144
+ '2. Analyse if the caption mathes described characters, wrap answer in <2nd_answer> tag.\n'+
145
+ '3. In case if there are any mismatches - rewrite caption to correct it wrapping '+
146
+ ' in <corrected_caption> tags. If the caption is fine - just write "no_need".',
147
+ }
148
+ ```
149
+
150
+ ## Grounding:
151
+
152
+ ```
153
+ grounding_prompt={
154
+ 'grounding_tags': ' Here are grounding tags for better understanding: ',
155
+ 'characters': ' Here is a list of characters that are present in the picture: ',
156
+ 'characters_traits': ' Here are popular tags or traits for each character on the picture: ',
157
+ 'grounding_info': ' Here is preliminary information about the picture: ',
158
+ 'no_chars': ' Do not use names for characters.',
159
+ }
160
+ ```
161
+
162
+ ## Composing userprompt:
163
+
164
+ After specifying selected mode, you can add prompt part for extra grounding and then privide it wrapping in corresponding xml tags:
165
+
166
+ `<tags>BOORU_TAGS</tags>.`
167
+
168
+ `<info>GENERAL_INFO</info>.`
169
+
170
+ `<characters>CHARACTER_NAMES</characters>.`
171
+
172
+ `<character_traits>CHARACTER1: [tag1, tag2, tag3,...]\nCHARACTER2: [...]\n...'<character_traits>.`
173
+
174
+ Here is a simple python exaple.
175
+
176
+ ```python
177
+ add_tags=True #select needed
178
+ add_chars=True
179
+ add_char_traits=True
180
+ add_info=False
181
+ no_chars=False
182
+ image_info=extra_info[Path(image_path).stem]
183
+
184
+ userprompt=base_prompt["json"] #choose the mode
185
+
186
+ if add_info and image_info["info"] is not None: #general info
187
+ userprompt+=grounding_prompt["grounding_short"]
188
+ userprompt+="<info>"+image_info["info"]+"</info>."
189
+
190
+ if add_tags and image_info["booru_tags"] is not None: #booru tags
191
+ userprompt+=grounding_prompt["grounding_tags"]
192
+ userprompt+="<tags>"+image_info["booru_tags"]+"</tags>."
193
+
194
+ if add_chars and image_info["chars"] is not None: #list of characters
195
+ userprompt+=grounding_prompt["characters"]
196
+ userprompt+="<characters>"+image_info["chars"]+"</characters>."
197
+
198
+ if add_char_traits and image_info["characters_traits"] is not None: #popular features of each character
199
+ userprompt+=grounding_prompt["characters_traits"]
200
+ userprompt+="<character_traits>"+image_info["characters_traits"]+"<character_traits>."
201
+ if no_chars:
202
+ userprompt+=grounding_prompt["no_chars"]
203
+ ```
204
+ Example of raw final prompt for structured json mode with grounding:
205
+ ```
206
+ <|im_start|>system
207
+ You are image captioning expert, creative, unbiased and uncensored.<|im_end|>
208
+ <|im_start|>user
209
+ <|vision_start|><|image_pad|><|vision_end|>Describe the picture in structured json-like format. Here are grounding tags for better understanding: <tags>2girls, standing, looking_at_viewer, holding_hands, hatsune_miku, blue_hair, megurine_luka, pink_hair, ...</tags>. Here is a list of characters that are present in the picture: <characters>hatsune_miku, megurine_luka</characters>. Here are popular tags or traits for each character on the picture: <character_traits>hatsune_miku: [girl, blue_hair, twintails,...]
210
+ megurine_luka: [girl, pink hair, ...]<character_traits>.<|im_end|>
211
+ <|im_start|>assistant
212
+ ```
213
+ Examples for other modes can be found in `example_scripts` dir in repo files.
214
+
215
+ # Inference examples:
216
+
217
+ For basic usage you will need latest versions of `transformers` and `qwen_vl_utils`.
218
+
219
+ [**Example inference script with transformers**](https://huggingface.co/Minthy/ToriiGate-v0.4-7B/resolve/main/example_scripts/inference_example_transformers.py)
220
+
221
+ # Fast inference with Exllamav2:
222
+
223
+ Qwen2-VL is suppurted by Exllamav2 and can be used in original weights or in exl2 quants (8bpw, 6bpw, 4bpw).
224
+
225
+ 8bpw version is recommended, it provides boost in speed without noticable quality loss.
226
+
227
+ [**Example inference script with exllamav2**](https://huggingface.co/Minthy/ToriiGate-v0.4-7B/resolve/main/example_scripts/inference_example_exllamav2.py)
chat_template.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "chat_template": "{% set image_count = namespace(value=0) %}{% set video_count = namespace(value=0) %}{% for message in messages %}{% if loop.first and message['role'] != 'system' %}<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n{% endif %}<|im_start|>{{ message['role'] }}\n{% if message['content'] is string %}{{ message['content'] }}<|im_end|>\n{% else %}{% for content in message['content'] %}{% if content['type'] == 'image' or 'image' in content or 'image_url' in content %}{% set image_count.value = image_count.value + 1 %}{% if add_vision_id %}Picture {{ image_count.value }}: {% endif %}<|vision_start|><|image_pad|><|vision_end|>{% elif content['type'] == 'video' or 'video' in content %}{% set video_count.value = video_count.value + 1 %}{% if add_vision_id %}Video {{ video_count.value }}: {% endif %}<|vision_start|><|video_pad|><|vision_end|>{% elif 'text' in content %}{{ content['text'] }}{% endif %}{% endfor %}<|im_end|>\n{% endif %}{% endfor %}{% if add_generation_prompt %}<|im_start|>assistant\n{% endif %}"
3
+ }
config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen2VLForConditionalGeneration"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "bos_token_id": 151643,
7
+ "eos_token_id": 151645,
8
+ "vision_start_token_id": 151652,
9
+ "vision_end_token_id": 151653,
10
+ "vision_token_id": 151654,
11
+ "image_token_id": 151655,
12
+ "video_token_id": 151656,
13
+ "hidden_act": "silu",
14
+ "hidden_size": 3584,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 18944,
17
+ "max_position_embeddings": 32768,
18
+ "max_window_layers": 28,
19
+ "model_type": "qwen2_vl",
20
+ "num_attention_heads": 28,
21
+ "num_hidden_layers": 28,
22
+ "num_key_value_heads": 4,
23
+ "rms_norm_eps": 1e-06,
24
+ "rope_theta": 1000000.0,
25
+ "sliding_window": 32768,
26
+ "tie_word_embeddings": false,
27
+ "torch_dtype": "bfloat16",
28
+ "transformers_version": "4.41.2",
29
+ "use_cache": true,
30
+ "use_sliding_window": false,
31
+ "vision_config": {
32
+ "depth": 32,
33
+ "embed_dim": 1280,
34
+ "mlp_ratio": 4,
35
+ "num_heads": 16,
36
+ "in_chans": 3,
37
+ "hidden_size": 3584,
38
+ "patch_size": 14,
39
+ "spatial_merge_size": 2,
40
+ "spatial_patch_size": 14,
41
+ "temporal_patch_size": 2
42
+ },
43
+ "rope_scaling": {
44
+ "type": "mrope",
45
+ "mrope_section": [
46
+ 16,
47
+ 24,
48
+ 24
49
+ ]
50
+ },
51
+ "vocab_size": 152064
52
+ }
example_scripts/inference_example_exllamav2.py ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from PIL import Image
3
+ from exllamav2 import (
4
+ ExLlamaV2,
5
+ ExLlamaV2Config,
6
+ ExLlamaV2Cache,
7
+ ExLlamaV2Tokenizer,
8
+ ExLlamaV2VisionTower,
9
+ )
10
+
11
+ from exllamav2.generator import (
12
+ ExLlamaV2DynamicGenerator,
13
+ ExLlamaV2Sampler,
14
+ )
15
+
16
+
17
+ model_id='./ToriiGate-v04-7b'
18
+ max_new_tokens=1000
19
+
20
+ image_file='/path/to/image_1.jpg'
21
+
22
+ image_info={}
23
+
24
+ image_info["booru_tags"]="2girls, standing, looking_at_viewer, holding_hands, hatsune_miku, blue_hair, megurine_luka, pink_hair, ..."
25
+ #image_info["booru_tags"]=open('/path/to/image_1_tags.txt').read().strip()
26
+ #image_info["booru_tags"]=None
27
+
28
+ image_info["chars"]="hatsune_miku, megurine_luka"
29
+ #image_info["chars"]=open('/path/to/image_1_char.txt').read().strip()
30
+ #image_info["chars"]=None
31
+
32
+ image_info["characters_traits"]="hatsune_miku: [girl, blue_hair, twintails,...]\nmegurine_luka: [girl, pink hair, ...]"
33
+ #image_info["characters_traits"]=open('/path/to/image_1_char_traits.txt').read().strip()
34
+ #image_info["characters_traits"]=None
35
+
36
+ image_info["info"]=None
37
+
38
+ base_prompt={
39
+ 'json': 'Describe the picture in structured json-like format.',
40
+ 'markdown': 'Describe the picture in structured markdown format.',
41
+ 'caption_vars': 'Write the following options for captions: ["Regular Summary","Individual Parts","Midjourney-Style Summary","DeviantArt Commission Request"].',
42
+ 'short': 'You need to write a medium-short and convenient caption for the picture.',
43
+ 'long': 'You need to write a long and very detailed caption for the picture.',
44
+ 'bbox': 'Write bounding boxes for each character and their faces.',
45
+ }
46
+
47
+ grounding_prompt={
48
+ 'grounding_tags': ' Here are grounding tags for better understanding: ',
49
+ 'characters': ' Here is a list of characters that are present in the picture: ',
50
+ 'characters_traits': ' Here are popular tags or traits for each character on the picture: ',
51
+ 'grounding_info': ' Here is preliminary information about the picture: ',
52
+ 'no_chars': ' Do not use names for characters.',
53
+ }
54
+
55
+ add_tags=True #select needed
56
+ add_chars=True
57
+ add_char_traits=True
58
+ add_info=False
59
+ no_chars=False
60
+
61
+ userprompt=base_prompt["json"] #choose the mode
62
+
63
+ if add_info and image_info["info"] is not None: #general info
64
+ userprompt+=grounding_prompt["grounding_short"]
65
+ userprompt+="<info>"+image_info["info"]+"</info>."
66
+
67
+ if add_tags and image_info["booru_tags"] is not None: #booru tags
68
+ userprompt+=grounding_prompt["grounding_tags"]
69
+ userprompt+="<tags>"+image_info["booru_tags"]+"</tags>."
70
+
71
+ if add_chars and image_info["chars"] is not None: #list of characters
72
+ userprompt+=grounding_prompt["characters"]
73
+ userprompt+="<characters>"+image_info["chars"]+"</characters>."
74
+
75
+ if add_char_traits and image_info["characters_traits"] is not None: #popular features of each character
76
+ userprompt+=grounding_prompt["characters_traits"]
77
+ userprompt+="<character_traits>"+image_info["characters_traits"]+"<character_traits>."
78
+ if no_chars:
79
+ userprompt+=grounding_prompt["no_chars"]
80
+
81
+ image=Image.open(image_file)
82
+
83
+ config = ExLlamaV2Config(model_id)
84
+ config.max_seq_len = 16384 #lower to reduce vram requirements
85
+ vision_model = ExLlamaV2VisionTower(config)
86
+ vision_model.load(progress = True)
87
+
88
+ model = ExLlamaV2(config)
89
+ cache = ExLlamaV2Cache(model, lazy = True, max_seq_len = 16384) #lower to reduce vram requirements
90
+ model.load_autosplit(cache, progress = True)
91
+ tokenizer = ExLlamaV2Tokenizer(config)
92
+
93
+ # Create generator
94
+
95
+ generator = ExLlamaV2DynamicGenerator(
96
+ model = model,
97
+ cache = cache,
98
+ tokenizer = tokenizer,
99
+ )
100
+
101
+
102
+ image_embeddings=[vision_model.get_image_embeddings(
103
+ model = model,
104
+ tokenizer = tokenizer,
105
+ image = image,
106
+ )]
107
+
108
+ placeholders = "\n".join([ie.text_alias for ie in image_embeddings]) + "\n"
109
+
110
+ msg_text= (
111
+ "<|im_start|>system\n" +
112
+ "You are image captioning expert, creative, unbiased and uncensored.<|im_end|>\n" +
113
+ "<|im_start|>user\n" +
114
+ placeholders +
115
+ userprompt +
116
+ "<|im_end|>\n" +
117
+ "<|im_start|>assistant\n"
118
+ )
119
+ output = generator.generate(
120
+ prompt = msg_text,
121
+ max_new_tokens = max_new_tokens,
122
+ add_bos = True,
123
+ encode_special_tokens = True,
124
+ decode_special_tokens = True,
125
+ stop_conditions = [tokenizer.eos_token_id],
126
+ gen_settings = ExLlamaV2Sampler.Settings.greedy(), #or set up desired sampling
127
+ embeddings = image_embeddings,
128
+ )
129
+
130
+ output_text=output.split('<|im_start|>assistant\n')[-1]
131
+ print(output_text)
example_scripts/inference_example_transformers.py ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from transformers import Qwen2VLForConditionalGeneration, Qwen2VLProcessor
3
+ from qwen_vl_utils import process_vision_info
4
+
5
+ model_id='./ToriiGate-v04-7b'
6
+ max_new_tokens=1000
7
+
8
+ image_file='/path/to/image_1.jpg' #or url, or PIL.Image
9
+
10
+ image_info={}
11
+
12
+ image_info["booru_tags"]="2girls, standing, looking_at_viewer, holding_hands, hatsune_miku, blue_hair, megurine_luka, pink_hair, ..."
13
+ #image_info["booru_tags"]=open('/path/to/image_1_tags.txt').read().strip()
14
+ #image_info["booru_tags"]=None
15
+
16
+ image_info["chars"]="hatsune_miku, megurine_luka"
17
+ #image_info["chars"]=open('/path/to/image_1_char.txt').read().strip()
18
+ #image_info["chars"]=None
19
+
20
+ image_info["characters_traits"]="hatsune_miku: [girl, blue_hair, twintails,...]\nmegurine_luka: [girl, pink hair, ...]"
21
+ #image_info["characters_traits"]=open('/path/to/image_1_char_traits.txt').read().strip()
22
+ #image_info["characters_traits"]=None
23
+
24
+ image_info["info"]=None
25
+
26
+ base_prompt={
27
+ 'json': 'Describe the picture in structured json-like format.',
28
+ 'markdown': 'Describe the picture in structured markdown format.',
29
+ 'caption_vars': 'Write the following options for captions: ["Regular Summary","Individual Parts","Midjourney-Style Summary","DeviantArt Commission Request"].',
30
+ 'short': 'You need to write a medium-short and convenient caption for the picture.',
31
+ 'long': 'You need to write a long and very detailed caption for the picture.',
32
+ 'bbox': 'Write bounding boxes for each character and their faces.',
33
+ 'check_and_correct': 'You need to compare given caption with the picture and given booru tags '+
34
+ ' using chain of thought.\n'+
35
+ '1. Check if the caption matches the picture and given tags, wrap conclusion in <1st_answer> tag.\n'+
36
+ '2. Analyse if the caption mathes described characters, wrap answer in <2nd_answer> tag.\n'+
37
+ '3. In case if there are any mismatches - rewrite caption to correct it wrapping '+
38
+ ' in <corrected_caption> tags. If the caption is fine - just write "no_need".',
39
+ }
40
+
41
+ grounding_prompt={
42
+ 'grounding_tags': ' Here are grounding tags for better understanding: ',
43
+ 'characters': ' Here is a list of characters that are present in the picture: ',
44
+ 'characters_traits': ' Here are popular tags or traits for each character on the picture: ',
45
+ 'grounding_info': ' Here is preliminary information about the picture: ',
46
+ 'no_chars': ' Do not use names for characters.',
47
+ }
48
+
49
+ add_tags=True #select needed
50
+ add_chars=True
51
+ add_char_traits=True
52
+ add_info=False
53
+ no_chars=False
54
+
55
+ userprompt=base_prompt["json"] #choose the mode
56
+
57
+ if add_info and image_info["info"] is not None: #general info
58
+ userprompt+=grounding_prompt["grounding_short"]
59
+ userprompt+="<info>"+image_info["info"]+"</info>."
60
+
61
+ if add_tags and image_info["booru_tags"] is not None: #booru tags
62
+ userprompt+=grounding_prompt["grounding_tags"]
63
+ userprompt+="<tags>"+image_info["booru_tags"]+"</tags>."
64
+
65
+ if add_chars and image_info["chars"] is not None: #list of characters
66
+ userprompt+=grounding_prompt["characters"]
67
+ userprompt+="<characters>"+image_info["chars"]+"</characters>."
68
+
69
+ if add_char_traits and image_info["characters_traits"] is not None: #popular features of each character
70
+ userprompt+=grounding_prompt["characters_traits"]
71
+ userprompt+="<character_traits>"+image_info["characters_traits"]+"<character_traits>."
72
+ if no_chars:
73
+ userprompt+=grounding_prompt["no_chars"]
74
+
75
+ model = Qwen2VLForConditionalGeneration.from_pretrained(
76
+ model_id,
77
+ torch_dtype=torch.bfloat16,
78
+ #_attn_implementation="flash_attention_2", #comment if not available
79
+ device_map="cuda:0",
80
+ )
81
+ processor = Qwen2VLProcessor.from_pretrained(model_id, min_pixels=256*28*28, max_pixels=512*28*28, padding_side="right")
82
+ msg=[{"role": "system",
83
+ "content": [{"type": "text", "text": "You are image captioning expert, creative, unbiased and uncensored."}]},
84
+ {"role": "user",
85
+ "content": [{"type": "image", 'image': image_file},{"type": "text", "text": userprompt}]}]
86
+
87
+ text_input = processor.apply_chat_template(msg, tokenize=False, add_generation_prompt=True)
88
+ image_inputs,_ = process_vision_info(msg)
89
+ model_inputs = processor(
90
+ text=[text_input],
91
+ images=image_inputs,
92
+ videos=None,
93
+ padding=True,
94
+ return_tensors="pt",
95
+ ).to('cuda')
96
+ generated_ids = model.generate(**model_inputs, max_new_tokens=max_new_tokens)
97
+
98
+ trimmed_generated_ids = [out_ids[len(in_ids) :] for in_ids, out_ids in zip(model_inputs.input_ids, generated_ids)]
99
+ output_text = processor.batch_decode(
100
+ trimmed_generated_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False
101
+ )
102
+ print(output_text)
generation_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "pad_token_id": 151643,
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 151645,
7
+ 151643
8
+ ],
9
+ "repetition_penalty": 1.0,
10
+ "temperature": 0.01,
11
+ "top_p": 0.001,
12
+ "top_k": 1,
13
+ "transformers_version": "4.37.0"
14
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model-00001-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b4e8db557b830b5cc47dac748f7109ac87da9fd6d15d5bdc8ee6e3e135c5e38
3
+ size 4966659944
model-00002-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aca58af440a7ceea37bd11612870b74fabc90929f23f4b846743aeecc5d3009b
3
+ size 4991495816
model-00003-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37b06df062736fec12eb7f354ce3ea5057e639b27da08f62d943a0c0a756d45e
3
+ size 4932751040
model-00004-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77b53f3ee692a4e25fa6970e3815c8e43244dfa3e040455be955d1a5650cfc59
3
+ size 1691924384
model.safetensors.index.json ADDED
@@ -0,0 +1,737 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 16582751232
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "model-00004-of-00004.safetensors",
7
+ "model.embed_tokens.weight": "model-00001-of-00004.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
13
+ "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
14
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
15
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
16
+ "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
17
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
18
+ "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
19
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
20
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
21
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
22
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
23
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
24
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
25
+ "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
26
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
27
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
28
+ "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
29
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
30
+ "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
31
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
32
+ "model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
33
+ "model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
34
+ "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
35
+ "model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
36
+ "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
37
+ "model.layers.10.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
38
+ "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
39
+ "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
40
+ "model.layers.10.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
41
+ "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
42
+ "model.layers.10.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
43
+ "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
44
+ "model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
45
+ "model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
46
+ "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
47
+ "model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
48
+ "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
49
+ "model.layers.11.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
50
+ "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
51
+ "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
52
+ "model.layers.11.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
53
+ "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
54
+ "model.layers.11.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
55
+ "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
56
+ "model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
57
+ "model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
58
+ "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
59
+ "model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
60
+ "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
61
+ "model.layers.12.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
62
+ "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
63
+ "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
64
+ "model.layers.12.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
65
+ "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
66
+ "model.layers.12.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
67
+ "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
68
+ "model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
69
+ "model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
70
+ "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
71
+ "model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
72
+ "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
73
+ "model.layers.13.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
74
+ "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
75
+ "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
76
+ "model.layers.13.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
77
+ "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
78
+ "model.layers.13.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
79
+ "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
80
+ "model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
81
+ "model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
82
+ "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
83
+ "model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
84
+ "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
85
+ "model.layers.14.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
86
+ "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
87
+ "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
88
+ "model.layers.14.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
89
+ "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
90
+ "model.layers.14.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
91
+ "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
92
+ "model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
93
+ "model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
94
+ "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
95
+ "model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
96
+ "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
97
+ "model.layers.15.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
98
+ "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
99
+ "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
100
+ "model.layers.15.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
101
+ "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
102
+ "model.layers.15.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
103
+ "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
104
+ "model.layers.16.input_layernorm.weight": "model-00003-of-00004.safetensors",
105
+ "model.layers.16.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
106
+ "model.layers.16.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
107
+ "model.layers.16.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
108
+ "model.layers.16.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
109
+ "model.layers.16.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
110
+ "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
111
+ "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
112
+ "model.layers.16.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
113
+ "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
114
+ "model.layers.16.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
115
+ "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
116
+ "model.layers.17.input_layernorm.weight": "model-00003-of-00004.safetensors",
117
+ "model.layers.17.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
118
+ "model.layers.17.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
119
+ "model.layers.17.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
120
+ "model.layers.17.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
121
+ "model.layers.17.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
122
+ "model.layers.17.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
123
+ "model.layers.17.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
124
+ "model.layers.17.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
125
+ "model.layers.17.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
126
+ "model.layers.17.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
127
+ "model.layers.17.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
128
+ "model.layers.18.input_layernorm.weight": "model-00003-of-00004.safetensors",
129
+ "model.layers.18.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
130
+ "model.layers.18.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
131
+ "model.layers.18.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
132
+ "model.layers.18.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
133
+ "model.layers.18.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
134
+ "model.layers.18.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
135
+ "model.layers.18.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
136
+ "model.layers.18.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
137
+ "model.layers.18.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
138
+ "model.layers.18.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
139
+ "model.layers.18.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
140
+ "model.layers.19.input_layernorm.weight": "model-00003-of-00004.safetensors",
141
+ "model.layers.19.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
142
+ "model.layers.19.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
143
+ "model.layers.19.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
144
+ "model.layers.19.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
145
+ "model.layers.19.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
146
+ "model.layers.19.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
147
+ "model.layers.19.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
148
+ "model.layers.19.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
149
+ "model.layers.19.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
150
+ "model.layers.19.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
151
+ "model.layers.19.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
152
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
153
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
154
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
155
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
156
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
157
+ "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
158
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
159
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
160
+ "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
161
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
162
+ "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
163
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
164
+ "model.layers.20.input_layernorm.weight": "model-00003-of-00004.safetensors",
165
+ "model.layers.20.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
166
+ "model.layers.20.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
167
+ "model.layers.20.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
168
+ "model.layers.20.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
169
+ "model.layers.20.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
170
+ "model.layers.20.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
171
+ "model.layers.20.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
172
+ "model.layers.20.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
173
+ "model.layers.20.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
174
+ "model.layers.20.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
175
+ "model.layers.20.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
176
+ "model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors",
177
+ "model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
178
+ "model.layers.21.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
179
+ "model.layers.21.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
180
+ "model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
181
+ "model.layers.21.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
182
+ "model.layers.21.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
183
+ "model.layers.21.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
184
+ "model.layers.21.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
185
+ "model.layers.21.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
186
+ "model.layers.21.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
187
+ "model.layers.21.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
188
+ "model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
189
+ "model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
190
+ "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
191
+ "model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
192
+ "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
193
+ "model.layers.22.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
194
+ "model.layers.22.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
195
+ "model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
196
+ "model.layers.22.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
197
+ "model.layers.22.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
198
+ "model.layers.22.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
199
+ "model.layers.22.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
200
+ "model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
201
+ "model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
202
+ "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
203
+ "model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
204
+ "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
205
+ "model.layers.23.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
206
+ "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
207
+ "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
208
+ "model.layers.23.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
209
+ "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
210
+ "model.layers.23.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
211
+ "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
212
+ "model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
213
+ "model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
214
+ "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
215
+ "model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
216
+ "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
217
+ "model.layers.24.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
218
+ "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
219
+ "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
220
+ "model.layers.24.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
221
+ "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
222
+ "model.layers.24.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
223
+ "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
224
+ "model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
225
+ "model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
226
+ "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
227
+ "model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
228
+ "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
229
+ "model.layers.25.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
230
+ "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
231
+ "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
232
+ "model.layers.25.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
233
+ "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
234
+ "model.layers.25.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
235
+ "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
236
+ "model.layers.26.input_layernorm.weight": "model-00004-of-00004.safetensors",
237
+ "model.layers.26.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
238
+ "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
239
+ "model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
240
+ "model.layers.26.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
241
+ "model.layers.26.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
242
+ "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
243
+ "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
244
+ "model.layers.26.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
245
+ "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
246
+ "model.layers.26.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
247
+ "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
248
+ "model.layers.27.input_layernorm.weight": "model-00004-of-00004.safetensors",
249
+ "model.layers.27.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
250
+ "model.layers.27.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
251
+ "model.layers.27.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
252
+ "model.layers.27.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
253
+ "model.layers.27.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
254
+ "model.layers.27.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
255
+ "model.layers.27.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
256
+ "model.layers.27.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
257
+ "model.layers.27.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
258
+ "model.layers.27.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
259
+ "model.layers.27.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
260
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
261
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
262
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
263
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
264
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
265
+ "model.layers.3.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
266
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
267
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
268
+ "model.layers.3.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
269
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
270
+ "model.layers.3.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
271
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
272
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
273
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
274
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
275
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
276
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
277
+ "model.layers.4.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
278
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
279
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
280
+ "model.layers.4.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
281
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
282
+ "model.layers.4.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
283
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
284
+ "model.layers.5.input_layernorm.weight": "model-00002-of-00004.safetensors",
285
+ "model.layers.5.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
286
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
287
+ "model.layers.5.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
288
+ "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
289
+ "model.layers.5.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
290
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
291
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
292
+ "model.layers.5.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
293
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
294
+ "model.layers.5.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
295
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
296
+ "model.layers.6.input_layernorm.weight": "model-00002-of-00004.safetensors",
297
+ "model.layers.6.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
298
+ "model.layers.6.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
299
+ "model.layers.6.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
300
+ "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
301
+ "model.layers.6.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
302
+ "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
303
+ "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
304
+ "model.layers.6.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
305
+ "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
306
+ "model.layers.6.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
307
+ "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
308
+ "model.layers.7.input_layernorm.weight": "model-00002-of-00004.safetensors",
309
+ "model.layers.7.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
310
+ "model.layers.7.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
311
+ "model.layers.7.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
312
+ "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
313
+ "model.layers.7.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
314
+ "model.layers.7.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
315
+ "model.layers.7.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
316
+ "model.layers.7.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
317
+ "model.layers.7.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
318
+ "model.layers.7.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
319
+ "model.layers.7.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
320
+ "model.layers.8.input_layernorm.weight": "model-00002-of-00004.safetensors",
321
+ "model.layers.8.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
322
+ "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
323
+ "model.layers.8.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
324
+ "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
325
+ "model.layers.8.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
326
+ "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
327
+ "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
328
+ "model.layers.8.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
329
+ "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
330
+ "model.layers.8.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
331
+ "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
332
+ "model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
333
+ "model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
334
+ "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
335
+ "model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
336
+ "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
337
+ "model.layers.9.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
338
+ "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
339
+ "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
340
+ "model.layers.9.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
341
+ "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
342
+ "model.layers.9.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
343
+ "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
344
+ "model.norm.weight": "model-00004-of-00004.safetensors",
345
+ "visual.blocks.0.attn.proj.bias": "model-00001-of-00004.safetensors",
346
+ "visual.blocks.0.attn.proj.weight": "model-00001-of-00004.safetensors",
347
+ "visual.blocks.0.attn.qkv.bias": "model-00001-of-00004.safetensors",
348
+ "visual.blocks.0.attn.qkv.weight": "model-00001-of-00004.safetensors",
349
+ "visual.blocks.0.mlp.fc1.bias": "model-00001-of-00004.safetensors",
350
+ "visual.blocks.0.mlp.fc1.weight": "model-00001-of-00004.safetensors",
351
+ "visual.blocks.0.mlp.fc2.bias": "model-00001-of-00004.safetensors",
352
+ "visual.blocks.0.mlp.fc2.weight": "model-00001-of-00004.safetensors",
353
+ "visual.blocks.0.norm1.bias": "model-00001-of-00004.safetensors",
354
+ "visual.blocks.0.norm1.weight": "model-00001-of-00004.safetensors",
355
+ "visual.blocks.0.norm2.bias": "model-00001-of-00004.safetensors",
356
+ "visual.blocks.0.norm2.weight": "model-00001-of-00004.safetensors",
357
+ "visual.blocks.1.attn.proj.bias": "model-00001-of-00004.safetensors",
358
+ "visual.blocks.1.attn.proj.weight": "model-00001-of-00004.safetensors",
359
+ "visual.blocks.1.attn.qkv.bias": "model-00001-of-00004.safetensors",
360
+ "visual.blocks.1.attn.qkv.weight": "model-00001-of-00004.safetensors",
361
+ "visual.blocks.1.mlp.fc1.bias": "model-00001-of-00004.safetensors",
362
+ "visual.blocks.1.mlp.fc1.weight": "model-00001-of-00004.safetensors",
363
+ "visual.blocks.1.mlp.fc2.bias": "model-00001-of-00004.safetensors",
364
+ "visual.blocks.1.mlp.fc2.weight": "model-00001-of-00004.safetensors",
365
+ "visual.blocks.1.norm1.bias": "model-00001-of-00004.safetensors",
366
+ "visual.blocks.1.norm1.weight": "model-00001-of-00004.safetensors",
367
+ "visual.blocks.1.norm2.bias": "model-00001-of-00004.safetensors",
368
+ "visual.blocks.1.norm2.weight": "model-00001-of-00004.safetensors",
369
+ "visual.blocks.10.attn.proj.bias": "model-00001-of-00004.safetensors",
370
+ "visual.blocks.10.attn.proj.weight": "model-00001-of-00004.safetensors",
371
+ "visual.blocks.10.attn.qkv.bias": "model-00001-of-00004.safetensors",
372
+ "visual.blocks.10.attn.qkv.weight": "model-00001-of-00004.safetensors",
373
+ "visual.blocks.10.mlp.fc1.bias": "model-00001-of-00004.safetensors",
374
+ "visual.blocks.10.mlp.fc1.weight": "model-00001-of-00004.safetensors",
375
+ "visual.blocks.10.mlp.fc2.bias": "model-00001-of-00004.safetensors",
376
+ "visual.blocks.10.mlp.fc2.weight": "model-00001-of-00004.safetensors",
377
+ "visual.blocks.10.norm1.bias": "model-00001-of-00004.safetensors",
378
+ "visual.blocks.10.norm1.weight": "model-00001-of-00004.safetensors",
379
+ "visual.blocks.10.norm2.bias": "model-00001-of-00004.safetensors",
380
+ "visual.blocks.10.norm2.weight": "model-00001-of-00004.safetensors",
381
+ "visual.blocks.11.attn.proj.bias": "model-00001-of-00004.safetensors",
382
+ "visual.blocks.11.attn.proj.weight": "model-00001-of-00004.safetensors",
383
+ "visual.blocks.11.attn.qkv.bias": "model-00001-of-00004.safetensors",
384
+ "visual.blocks.11.attn.qkv.weight": "model-00001-of-00004.safetensors",
385
+ "visual.blocks.11.mlp.fc1.bias": "model-00001-of-00004.safetensors",
386
+ "visual.blocks.11.mlp.fc1.weight": "model-00001-of-00004.safetensors",
387
+ "visual.blocks.11.mlp.fc2.bias": "model-00001-of-00004.safetensors",
388
+ "visual.blocks.11.mlp.fc2.weight": "model-00001-of-00004.safetensors",
389
+ "visual.blocks.11.norm1.bias": "model-00001-of-00004.safetensors",
390
+ "visual.blocks.11.norm1.weight": "model-00001-of-00004.safetensors",
391
+ "visual.blocks.11.norm2.bias": "model-00001-of-00004.safetensors",
392
+ "visual.blocks.11.norm2.weight": "model-00001-of-00004.safetensors",
393
+ "visual.blocks.12.attn.proj.bias": "model-00001-of-00004.safetensors",
394
+ "visual.blocks.12.attn.proj.weight": "model-00001-of-00004.safetensors",
395
+ "visual.blocks.12.attn.qkv.bias": "model-00001-of-00004.safetensors",
396
+ "visual.blocks.12.attn.qkv.weight": "model-00001-of-00004.safetensors",
397
+ "visual.blocks.12.mlp.fc1.bias": "model-00001-of-00004.safetensors",
398
+ "visual.blocks.12.mlp.fc1.weight": "model-00001-of-00004.safetensors",
399
+ "visual.blocks.12.mlp.fc2.bias": "model-00001-of-00004.safetensors",
400
+ "visual.blocks.12.mlp.fc2.weight": "model-00001-of-00004.safetensors",
401
+ "visual.blocks.12.norm1.bias": "model-00001-of-00004.safetensors",
402
+ "visual.blocks.12.norm1.weight": "model-00001-of-00004.safetensors",
403
+ "visual.blocks.12.norm2.bias": "model-00001-of-00004.safetensors",
404
+ "visual.blocks.12.norm2.weight": "model-00001-of-00004.safetensors",
405
+ "visual.blocks.13.attn.proj.bias": "model-00001-of-00004.safetensors",
406
+ "visual.blocks.13.attn.proj.weight": "model-00001-of-00004.safetensors",
407
+ "visual.blocks.13.attn.qkv.bias": "model-00001-of-00004.safetensors",
408
+ "visual.blocks.13.attn.qkv.weight": "model-00001-of-00004.safetensors",
409
+ "visual.blocks.13.mlp.fc1.bias": "model-00001-of-00004.safetensors",
410
+ "visual.blocks.13.mlp.fc1.weight": "model-00001-of-00004.safetensors",
411
+ "visual.blocks.13.mlp.fc2.bias": "model-00001-of-00004.safetensors",
412
+ "visual.blocks.13.mlp.fc2.weight": "model-00001-of-00004.safetensors",
413
+ "visual.blocks.13.norm1.bias": "model-00001-of-00004.safetensors",
414
+ "visual.blocks.13.norm1.weight": "model-00001-of-00004.safetensors",
415
+ "visual.blocks.13.norm2.bias": "model-00001-of-00004.safetensors",
416
+ "visual.blocks.13.norm2.weight": "model-00001-of-00004.safetensors",
417
+ "visual.blocks.14.attn.proj.bias": "model-00001-of-00004.safetensors",
418
+ "visual.blocks.14.attn.proj.weight": "model-00001-of-00004.safetensors",
419
+ "visual.blocks.14.attn.qkv.bias": "model-00001-of-00004.safetensors",
420
+ "visual.blocks.14.attn.qkv.weight": "model-00001-of-00004.safetensors",
421
+ "visual.blocks.14.mlp.fc1.bias": "model-00001-of-00004.safetensors",
422
+ "visual.blocks.14.mlp.fc1.weight": "model-00001-of-00004.safetensors",
423
+ "visual.blocks.14.mlp.fc2.bias": "model-00001-of-00004.safetensors",
424
+ "visual.blocks.14.mlp.fc2.weight": "model-00001-of-00004.safetensors",
425
+ "visual.blocks.14.norm1.bias": "model-00001-of-00004.safetensors",
426
+ "visual.blocks.14.norm1.weight": "model-00001-of-00004.safetensors",
427
+ "visual.blocks.14.norm2.bias": "model-00001-of-00004.safetensors",
428
+ "visual.blocks.14.norm2.weight": "model-00001-of-00004.safetensors",
429
+ "visual.blocks.15.attn.proj.bias": "model-00001-of-00004.safetensors",
430
+ "visual.blocks.15.attn.proj.weight": "model-00001-of-00004.safetensors",
431
+ "visual.blocks.15.attn.qkv.bias": "model-00001-of-00004.safetensors",
432
+ "visual.blocks.15.attn.qkv.weight": "model-00001-of-00004.safetensors",
433
+ "visual.blocks.15.mlp.fc1.bias": "model-00001-of-00004.safetensors",
434
+ "visual.blocks.15.mlp.fc1.weight": "model-00001-of-00004.safetensors",
435
+ "visual.blocks.15.mlp.fc2.bias": "model-00001-of-00004.safetensors",
436
+ "visual.blocks.15.mlp.fc2.weight": "model-00001-of-00004.safetensors",
437
+ "visual.blocks.15.norm1.bias": "model-00001-of-00004.safetensors",
438
+ "visual.blocks.15.norm1.weight": "model-00001-of-00004.safetensors",
439
+ "visual.blocks.15.norm2.bias": "model-00001-of-00004.safetensors",
440
+ "visual.blocks.15.norm2.weight": "model-00001-of-00004.safetensors",
441
+ "visual.blocks.16.attn.proj.bias": "model-00001-of-00004.safetensors",
442
+ "visual.blocks.16.attn.proj.weight": "model-00001-of-00004.safetensors",
443
+ "visual.blocks.16.attn.qkv.bias": "model-00001-of-00004.safetensors",
444
+ "visual.blocks.16.attn.qkv.weight": "model-00001-of-00004.safetensors",
445
+ "visual.blocks.16.mlp.fc1.bias": "model-00001-of-00004.safetensors",
446
+ "visual.blocks.16.mlp.fc1.weight": "model-00001-of-00004.safetensors",
447
+ "visual.blocks.16.mlp.fc2.bias": "model-00001-of-00004.safetensors",
448
+ "visual.blocks.16.mlp.fc2.weight": "model-00001-of-00004.safetensors",
449
+ "visual.blocks.16.norm1.bias": "model-00001-of-00004.safetensors",
450
+ "visual.blocks.16.norm1.weight": "model-00001-of-00004.safetensors",
451
+ "visual.blocks.16.norm2.bias": "model-00001-of-00004.safetensors",
452
+ "visual.blocks.16.norm2.weight": "model-00001-of-00004.safetensors",
453
+ "visual.blocks.17.attn.proj.bias": "model-00001-of-00004.safetensors",
454
+ "visual.blocks.17.attn.proj.weight": "model-00001-of-00004.safetensors",
455
+ "visual.blocks.17.attn.qkv.bias": "model-00001-of-00004.safetensors",
456
+ "visual.blocks.17.attn.qkv.weight": "model-00001-of-00004.safetensors",
457
+ "visual.blocks.17.mlp.fc1.bias": "model-00001-of-00004.safetensors",
458
+ "visual.blocks.17.mlp.fc1.weight": "model-00001-of-00004.safetensors",
459
+ "visual.blocks.17.mlp.fc2.bias": "model-00001-of-00004.safetensors",
460
+ "visual.blocks.17.mlp.fc2.weight": "model-00001-of-00004.safetensors",
461
+ "visual.blocks.17.norm1.bias": "model-00001-of-00004.safetensors",
462
+ "visual.blocks.17.norm1.weight": "model-00001-of-00004.safetensors",
463
+ "visual.blocks.17.norm2.bias": "model-00001-of-00004.safetensors",
464
+ "visual.blocks.17.norm2.weight": "model-00001-of-00004.safetensors",
465
+ "visual.blocks.18.attn.proj.bias": "model-00001-of-00004.safetensors",
466
+ "visual.blocks.18.attn.proj.weight": "model-00001-of-00004.safetensors",
467
+ "visual.blocks.18.attn.qkv.bias": "model-00001-of-00004.safetensors",
468
+ "visual.blocks.18.attn.qkv.weight": "model-00001-of-00004.safetensors",
469
+ "visual.blocks.18.mlp.fc1.bias": "model-00001-of-00004.safetensors",
470
+ "visual.blocks.18.mlp.fc1.weight": "model-00001-of-00004.safetensors",
471
+ "visual.blocks.18.mlp.fc2.bias": "model-00001-of-00004.safetensors",
472
+ "visual.blocks.18.mlp.fc2.weight": "model-00001-of-00004.safetensors",
473
+ "visual.blocks.18.norm1.bias": "model-00001-of-00004.safetensors",
474
+ "visual.blocks.18.norm1.weight": "model-00001-of-00004.safetensors",
475
+ "visual.blocks.18.norm2.bias": "model-00001-of-00004.safetensors",
476
+ "visual.blocks.18.norm2.weight": "model-00001-of-00004.safetensors",
477
+ "visual.blocks.19.attn.proj.bias": "model-00001-of-00004.safetensors",
478
+ "visual.blocks.19.attn.proj.weight": "model-00001-of-00004.safetensors",
479
+ "visual.blocks.19.attn.qkv.bias": "model-00001-of-00004.safetensors",
480
+ "visual.blocks.19.attn.qkv.weight": "model-00001-of-00004.safetensors",
481
+ "visual.blocks.19.mlp.fc1.bias": "model-00001-of-00004.safetensors",
482
+ "visual.blocks.19.mlp.fc1.weight": "model-00001-of-00004.safetensors",
483
+ "visual.blocks.19.mlp.fc2.bias": "model-00001-of-00004.safetensors",
484
+ "visual.blocks.19.mlp.fc2.weight": "model-00001-of-00004.safetensors",
485
+ "visual.blocks.19.norm1.bias": "model-00001-of-00004.safetensors",
486
+ "visual.blocks.19.norm1.weight": "model-00001-of-00004.safetensors",
487
+ "visual.blocks.19.norm2.bias": "model-00001-of-00004.safetensors",
488
+ "visual.blocks.19.norm2.weight": "model-00001-of-00004.safetensors",
489
+ "visual.blocks.2.attn.proj.bias": "model-00001-of-00004.safetensors",
490
+ "visual.blocks.2.attn.proj.weight": "model-00001-of-00004.safetensors",
491
+ "visual.blocks.2.attn.qkv.bias": "model-00001-of-00004.safetensors",
492
+ "visual.blocks.2.attn.qkv.weight": "model-00001-of-00004.safetensors",
493
+ "visual.blocks.2.mlp.fc1.bias": "model-00001-of-00004.safetensors",
494
+ "visual.blocks.2.mlp.fc1.weight": "model-00001-of-00004.safetensors",
495
+ "visual.blocks.2.mlp.fc2.bias": "model-00001-of-00004.safetensors",
496
+ "visual.blocks.2.mlp.fc2.weight": "model-00001-of-00004.safetensors",
497
+ "visual.blocks.2.norm1.bias": "model-00001-of-00004.safetensors",
498
+ "visual.blocks.2.norm1.weight": "model-00001-of-00004.safetensors",
499
+ "visual.blocks.2.norm2.bias": "model-00001-of-00004.safetensors",
500
+ "visual.blocks.2.norm2.weight": "model-00001-of-00004.safetensors",
501
+ "visual.blocks.20.attn.proj.bias": "model-00001-of-00004.safetensors",
502
+ "visual.blocks.20.attn.proj.weight": "model-00001-of-00004.safetensors",
503
+ "visual.blocks.20.attn.qkv.bias": "model-00001-of-00004.safetensors",
504
+ "visual.blocks.20.attn.qkv.weight": "model-00001-of-00004.safetensors",
505
+ "visual.blocks.20.mlp.fc1.bias": "model-00001-of-00004.safetensors",
506
+ "visual.blocks.20.mlp.fc1.weight": "model-00001-of-00004.safetensors",
507
+ "visual.blocks.20.mlp.fc2.bias": "model-00001-of-00004.safetensors",
508
+ "visual.blocks.20.mlp.fc2.weight": "model-00001-of-00004.safetensors",
509
+ "visual.blocks.20.norm1.bias": "model-00001-of-00004.safetensors",
510
+ "visual.blocks.20.norm1.weight": "model-00001-of-00004.safetensors",
511
+ "visual.blocks.20.norm2.bias": "model-00001-of-00004.safetensors",
512
+ "visual.blocks.20.norm2.weight": "model-00001-of-00004.safetensors",
513
+ "visual.blocks.21.attn.proj.bias": "model-00001-of-00004.safetensors",
514
+ "visual.blocks.21.attn.proj.weight": "model-00001-of-00004.safetensors",
515
+ "visual.blocks.21.attn.qkv.bias": "model-00001-of-00004.safetensors",
516
+ "visual.blocks.21.attn.qkv.weight": "model-00001-of-00004.safetensors",
517
+ "visual.blocks.21.mlp.fc1.bias": "model-00001-of-00004.safetensors",
518
+ "visual.blocks.21.mlp.fc1.weight": "model-00001-of-00004.safetensors",
519
+ "visual.blocks.21.mlp.fc2.bias": "model-00001-of-00004.safetensors",
520
+ "visual.blocks.21.mlp.fc2.weight": "model-00001-of-00004.safetensors",
521
+ "visual.blocks.21.norm1.bias": "model-00001-of-00004.safetensors",
522
+ "visual.blocks.21.norm1.weight": "model-00001-of-00004.safetensors",
523
+ "visual.blocks.21.norm2.bias": "model-00001-of-00004.safetensors",
524
+ "visual.blocks.21.norm2.weight": "model-00001-of-00004.safetensors",
525
+ "visual.blocks.22.attn.proj.bias": "model-00001-of-00004.safetensors",
526
+ "visual.blocks.22.attn.proj.weight": "model-00001-of-00004.safetensors",
527
+ "visual.blocks.22.attn.qkv.bias": "model-00001-of-00004.safetensors",
528
+ "visual.blocks.22.attn.qkv.weight": "model-00001-of-00004.safetensors",
529
+ "visual.blocks.22.mlp.fc1.bias": "model-00001-of-00004.safetensors",
530
+ "visual.blocks.22.mlp.fc1.weight": "model-00001-of-00004.safetensors",
531
+ "visual.blocks.22.mlp.fc2.bias": "model-00001-of-00004.safetensors",
532
+ "visual.blocks.22.mlp.fc2.weight": "model-00001-of-00004.safetensors",
533
+ "visual.blocks.22.norm1.bias": "model-00001-of-00004.safetensors",
534
+ "visual.blocks.22.norm1.weight": "model-00001-of-00004.safetensors",
535
+ "visual.blocks.22.norm2.bias": "model-00001-of-00004.safetensors",
536
+ "visual.blocks.22.norm2.weight": "model-00001-of-00004.safetensors",
537
+ "visual.blocks.23.attn.proj.bias": "model-00001-of-00004.safetensors",
538
+ "visual.blocks.23.attn.proj.weight": "model-00001-of-00004.safetensors",
539
+ "visual.blocks.23.attn.qkv.bias": "model-00001-of-00004.safetensors",
540
+ "visual.blocks.23.attn.qkv.weight": "model-00001-of-00004.safetensors",
541
+ "visual.blocks.23.mlp.fc1.bias": "model-00001-of-00004.safetensors",
542
+ "visual.blocks.23.mlp.fc1.weight": "model-00001-of-00004.safetensors",
543
+ "visual.blocks.23.mlp.fc2.bias": "model-00001-of-00004.safetensors",
544
+ "visual.blocks.23.mlp.fc2.weight": "model-00001-of-00004.safetensors",
545
+ "visual.blocks.23.norm1.bias": "model-00001-of-00004.safetensors",
546
+ "visual.blocks.23.norm1.weight": "model-00001-of-00004.safetensors",
547
+ "visual.blocks.23.norm2.bias": "model-00001-of-00004.safetensors",
548
+ "visual.blocks.23.norm2.weight": "model-00001-of-00004.safetensors",
549
+ "visual.blocks.24.attn.proj.bias": "model-00001-of-00004.safetensors",
550
+ "visual.blocks.24.attn.proj.weight": "model-00001-of-00004.safetensors",
551
+ "visual.blocks.24.attn.qkv.bias": "model-00001-of-00004.safetensors",
552
+ "visual.blocks.24.attn.qkv.weight": "model-00001-of-00004.safetensors",
553
+ "visual.blocks.24.mlp.fc1.bias": "model-00001-of-00004.safetensors",
554
+ "visual.blocks.24.mlp.fc1.weight": "model-00001-of-00004.safetensors",
555
+ "visual.blocks.24.mlp.fc2.bias": "model-00001-of-00004.safetensors",
556
+ "visual.blocks.24.mlp.fc2.weight": "model-00001-of-00004.safetensors",
557
+ "visual.blocks.24.norm1.bias": "model-00001-of-00004.safetensors",
558
+ "visual.blocks.24.norm1.weight": "model-00001-of-00004.safetensors",
559
+ "visual.blocks.24.norm2.bias": "model-00001-of-00004.safetensors",
560
+ "visual.blocks.24.norm2.weight": "model-00001-of-00004.safetensors",
561
+ "visual.blocks.25.attn.proj.bias": "model-00001-of-00004.safetensors",
562
+ "visual.blocks.25.attn.proj.weight": "model-00001-of-00004.safetensors",
563
+ "visual.blocks.25.attn.qkv.bias": "model-00001-of-00004.safetensors",
564
+ "visual.blocks.25.attn.qkv.weight": "model-00001-of-00004.safetensors",
565
+ "visual.blocks.25.mlp.fc1.bias": "model-00001-of-00004.safetensors",
566
+ "visual.blocks.25.mlp.fc1.weight": "model-00001-of-00004.safetensors",
567
+ "visual.blocks.25.mlp.fc2.bias": "model-00001-of-00004.safetensors",
568
+ "visual.blocks.25.mlp.fc2.weight": "model-00001-of-00004.safetensors",
569
+ "visual.blocks.25.norm1.bias": "model-00001-of-00004.safetensors",
570
+ "visual.blocks.25.norm1.weight": "model-00001-of-00004.safetensors",
571
+ "visual.blocks.25.norm2.bias": "model-00001-of-00004.safetensors",
572
+ "visual.blocks.25.norm2.weight": "model-00001-of-00004.safetensors",
573
+ "visual.blocks.26.attn.proj.bias": "model-00001-of-00004.safetensors",
574
+ "visual.blocks.26.attn.proj.weight": "model-00001-of-00004.safetensors",
575
+ "visual.blocks.26.attn.qkv.bias": "model-00001-of-00004.safetensors",
576
+ "visual.blocks.26.attn.qkv.weight": "model-00001-of-00004.safetensors",
577
+ "visual.blocks.26.mlp.fc1.bias": "model-00001-of-00004.safetensors",
578
+ "visual.blocks.26.mlp.fc1.weight": "model-00001-of-00004.safetensors",
579
+ "visual.blocks.26.mlp.fc2.bias": "model-00001-of-00004.safetensors",
580
+ "visual.blocks.26.mlp.fc2.weight": "model-00001-of-00004.safetensors",
581
+ "visual.blocks.26.norm1.bias": "model-00001-of-00004.safetensors",
582
+ "visual.blocks.26.norm1.weight": "model-00001-of-00004.safetensors",
583
+ "visual.blocks.26.norm2.bias": "model-00001-of-00004.safetensors",
584
+ "visual.blocks.26.norm2.weight": "model-00001-of-00004.safetensors",
585
+ "visual.blocks.27.attn.proj.bias": "model-00001-of-00004.safetensors",
586
+ "visual.blocks.27.attn.proj.weight": "model-00001-of-00004.safetensors",
587
+ "visual.blocks.27.attn.qkv.bias": "model-00001-of-00004.safetensors",
588
+ "visual.blocks.27.attn.qkv.weight": "model-00001-of-00004.safetensors",
589
+ "visual.blocks.27.mlp.fc1.bias": "model-00001-of-00004.safetensors",
590
+ "visual.blocks.27.mlp.fc1.weight": "model-00001-of-00004.safetensors",
591
+ "visual.blocks.27.mlp.fc2.bias": "model-00001-of-00004.safetensors",
592
+ "visual.blocks.27.mlp.fc2.weight": "model-00001-of-00004.safetensors",
593
+ "visual.blocks.27.norm1.bias": "model-00001-of-00004.safetensors",
594
+ "visual.blocks.27.norm1.weight": "model-00001-of-00004.safetensors",
595
+ "visual.blocks.27.norm2.bias": "model-00001-of-00004.safetensors",
596
+ "visual.blocks.27.norm2.weight": "model-00001-of-00004.safetensors",
597
+ "visual.blocks.28.attn.proj.bias": "model-00001-of-00004.safetensors",
598
+ "visual.blocks.28.attn.proj.weight": "model-00001-of-00004.safetensors",
599
+ "visual.blocks.28.attn.qkv.bias": "model-00001-of-00004.safetensors",
600
+ "visual.blocks.28.attn.qkv.weight": "model-00001-of-00004.safetensors",
601
+ "visual.blocks.28.mlp.fc1.bias": "model-00001-of-00004.safetensors",
602
+ "visual.blocks.28.mlp.fc1.weight": "model-00001-of-00004.safetensors",
603
+ "visual.blocks.28.mlp.fc2.bias": "model-00001-of-00004.safetensors",
604
+ "visual.blocks.28.mlp.fc2.weight": "model-00001-of-00004.safetensors",
605
+ "visual.blocks.28.norm1.bias": "model-00001-of-00004.safetensors",
606
+ "visual.blocks.28.norm1.weight": "model-00001-of-00004.safetensors",
607
+ "visual.blocks.28.norm2.bias": "model-00001-of-00004.safetensors",
608
+ "visual.blocks.28.norm2.weight": "model-00001-of-00004.safetensors",
609
+ "visual.blocks.29.attn.proj.bias": "model-00001-of-00004.safetensors",
610
+ "visual.blocks.29.attn.proj.weight": "model-00001-of-00004.safetensors",
611
+ "visual.blocks.29.attn.qkv.bias": "model-00001-of-00004.safetensors",
612
+ "visual.blocks.29.attn.qkv.weight": "model-00001-of-00004.safetensors",
613
+ "visual.blocks.29.mlp.fc1.bias": "model-00001-of-00004.safetensors",
614
+ "visual.blocks.29.mlp.fc1.weight": "model-00001-of-00004.safetensors",
615
+ "visual.blocks.29.mlp.fc2.bias": "model-00001-of-00004.safetensors",
616
+ "visual.blocks.29.mlp.fc2.weight": "model-00001-of-00004.safetensors",
617
+ "visual.blocks.29.norm1.bias": "model-00001-of-00004.safetensors",
618
+ "visual.blocks.29.norm1.weight": "model-00001-of-00004.safetensors",
619
+ "visual.blocks.29.norm2.bias": "model-00001-of-00004.safetensors",
620
+ "visual.blocks.29.norm2.weight": "model-00001-of-00004.safetensors",
621
+ "visual.blocks.3.attn.proj.bias": "model-00001-of-00004.safetensors",
622
+ "visual.blocks.3.attn.proj.weight": "model-00001-of-00004.safetensors",
623
+ "visual.blocks.3.attn.qkv.bias": "model-00001-of-00004.safetensors",
624
+ "visual.blocks.3.attn.qkv.weight": "model-00001-of-00004.safetensors",
625
+ "visual.blocks.3.mlp.fc1.bias": "model-00001-of-00004.safetensors",
626
+ "visual.blocks.3.mlp.fc1.weight": "model-00001-of-00004.safetensors",
627
+ "visual.blocks.3.mlp.fc2.bias": "model-00001-of-00004.safetensors",
628
+ "visual.blocks.3.mlp.fc2.weight": "model-00001-of-00004.safetensors",
629
+ "visual.blocks.3.norm1.bias": "model-00001-of-00004.safetensors",
630
+ "visual.blocks.3.norm1.weight": "model-00001-of-00004.safetensors",
631
+ "visual.blocks.3.norm2.bias": "model-00001-of-00004.safetensors",
632
+ "visual.blocks.3.norm2.weight": "model-00001-of-00004.safetensors",
633
+ "visual.blocks.30.attn.proj.bias": "model-00001-of-00004.safetensors",
634
+ "visual.blocks.30.attn.proj.weight": "model-00001-of-00004.safetensors",
635
+ "visual.blocks.30.attn.qkv.bias": "model-00001-of-00004.safetensors",
636
+ "visual.blocks.30.attn.qkv.weight": "model-00001-of-00004.safetensors",
637
+ "visual.blocks.30.mlp.fc1.bias": "model-00001-of-00004.safetensors",
638
+ "visual.blocks.30.mlp.fc1.weight": "model-00001-of-00004.safetensors",
639
+ "visual.blocks.30.mlp.fc2.bias": "model-00001-of-00004.safetensors",
640
+ "visual.blocks.30.mlp.fc2.weight": "model-00001-of-00004.safetensors",
641
+ "visual.blocks.30.norm1.bias": "model-00001-of-00004.safetensors",
642
+ "visual.blocks.30.norm1.weight": "model-00001-of-00004.safetensors",
643
+ "visual.blocks.30.norm2.bias": "model-00001-of-00004.safetensors",
644
+ "visual.blocks.30.norm2.weight": "model-00001-of-00004.safetensors",
645
+ "visual.blocks.31.attn.proj.bias": "model-00001-of-00004.safetensors",
646
+ "visual.blocks.31.attn.proj.weight": "model-00001-of-00004.safetensors",
647
+ "visual.blocks.31.attn.qkv.bias": "model-00001-of-00004.safetensors",
648
+ "visual.blocks.31.attn.qkv.weight": "model-00001-of-00004.safetensors",
649
+ "visual.blocks.31.mlp.fc1.bias": "model-00001-of-00004.safetensors",
650
+ "visual.blocks.31.mlp.fc1.weight": "model-00001-of-00004.safetensors",
651
+ "visual.blocks.31.mlp.fc2.bias": "model-00001-of-00004.safetensors",
652
+ "visual.blocks.31.mlp.fc2.weight": "model-00001-of-00004.safetensors",
653
+ "visual.blocks.31.norm1.bias": "model-00001-of-00004.safetensors",
654
+ "visual.blocks.31.norm1.weight": "model-00001-of-00004.safetensors",
655
+ "visual.blocks.31.norm2.bias": "model-00001-of-00004.safetensors",
656
+ "visual.blocks.31.norm2.weight": "model-00001-of-00004.safetensors",
657
+ "visual.blocks.4.attn.proj.bias": "model-00001-of-00004.safetensors",
658
+ "visual.blocks.4.attn.proj.weight": "model-00001-of-00004.safetensors",
659
+ "visual.blocks.4.attn.qkv.bias": "model-00001-of-00004.safetensors",
660
+ "visual.blocks.4.attn.qkv.weight": "model-00001-of-00004.safetensors",
661
+ "visual.blocks.4.mlp.fc1.bias": "model-00001-of-00004.safetensors",
662
+ "visual.blocks.4.mlp.fc1.weight": "model-00001-of-00004.safetensors",
663
+ "visual.blocks.4.mlp.fc2.bias": "model-00001-of-00004.safetensors",
664
+ "visual.blocks.4.mlp.fc2.weight": "model-00001-of-00004.safetensors",
665
+ "visual.blocks.4.norm1.bias": "model-00001-of-00004.safetensors",
666
+ "visual.blocks.4.norm1.weight": "model-00001-of-00004.safetensors",
667
+ "visual.blocks.4.norm2.bias": "model-00001-of-00004.safetensors",
668
+ "visual.blocks.4.norm2.weight": "model-00001-of-00004.safetensors",
669
+ "visual.blocks.5.attn.proj.bias": "model-00001-of-00004.safetensors",
670
+ "visual.blocks.5.attn.proj.weight": "model-00001-of-00004.safetensors",
671
+ "visual.blocks.5.attn.qkv.bias": "model-00001-of-00004.safetensors",
672
+ "visual.blocks.5.attn.qkv.weight": "model-00001-of-00004.safetensors",
673
+ "visual.blocks.5.mlp.fc1.bias": "model-00001-of-00004.safetensors",
674
+ "visual.blocks.5.mlp.fc1.weight": "model-00001-of-00004.safetensors",
675
+ "visual.blocks.5.mlp.fc2.bias": "model-00001-of-00004.safetensors",
676
+ "visual.blocks.5.mlp.fc2.weight": "model-00001-of-00004.safetensors",
677
+ "visual.blocks.5.norm1.bias": "model-00001-of-00004.safetensors",
678
+ "visual.blocks.5.norm1.weight": "model-00001-of-00004.safetensors",
679
+ "visual.blocks.5.norm2.bias": "model-00001-of-00004.safetensors",
680
+ "visual.blocks.5.norm2.weight": "model-00001-of-00004.safetensors",
681
+ "visual.blocks.6.attn.proj.bias": "model-00001-of-00004.safetensors",
682
+ "visual.blocks.6.attn.proj.weight": "model-00001-of-00004.safetensors",
683
+ "visual.blocks.6.attn.qkv.bias": "model-00001-of-00004.safetensors",
684
+ "visual.blocks.6.attn.qkv.weight": "model-00001-of-00004.safetensors",
685
+ "visual.blocks.6.mlp.fc1.bias": "model-00001-of-00004.safetensors",
686
+ "visual.blocks.6.mlp.fc1.weight": "model-00001-of-00004.safetensors",
687
+ "visual.blocks.6.mlp.fc2.bias": "model-00001-of-00004.safetensors",
688
+ "visual.blocks.6.mlp.fc2.weight": "model-00001-of-00004.safetensors",
689
+ "visual.blocks.6.norm1.bias": "model-00001-of-00004.safetensors",
690
+ "visual.blocks.6.norm1.weight": "model-00001-of-00004.safetensors",
691
+ "visual.blocks.6.norm2.bias": "model-00001-of-00004.safetensors",
692
+ "visual.blocks.6.norm2.weight": "model-00001-of-00004.safetensors",
693
+ "visual.blocks.7.attn.proj.bias": "model-00001-of-00004.safetensors",
694
+ "visual.blocks.7.attn.proj.weight": "model-00001-of-00004.safetensors",
695
+ "visual.blocks.7.attn.qkv.bias": "model-00001-of-00004.safetensors",
696
+ "visual.blocks.7.attn.qkv.weight": "model-00001-of-00004.safetensors",
697
+ "visual.blocks.7.mlp.fc1.bias": "model-00001-of-00004.safetensors",
698
+ "visual.blocks.7.mlp.fc1.weight": "model-00001-of-00004.safetensors",
699
+ "visual.blocks.7.mlp.fc2.bias": "model-00001-of-00004.safetensors",
700
+ "visual.blocks.7.mlp.fc2.weight": "model-00001-of-00004.safetensors",
701
+ "visual.blocks.7.norm1.bias": "model-00001-of-00004.safetensors",
702
+ "visual.blocks.7.norm1.weight": "model-00001-of-00004.safetensors",
703
+ "visual.blocks.7.norm2.bias": "model-00001-of-00004.safetensors",
704
+ "visual.blocks.7.norm2.weight": "model-00001-of-00004.safetensors",
705
+ "visual.blocks.8.attn.proj.bias": "model-00001-of-00004.safetensors",
706
+ "visual.blocks.8.attn.proj.weight": "model-00001-of-00004.safetensors",
707
+ "visual.blocks.8.attn.qkv.bias": "model-00001-of-00004.safetensors",
708
+ "visual.blocks.8.attn.qkv.weight": "model-00001-of-00004.safetensors",
709
+ "visual.blocks.8.mlp.fc1.bias": "model-00001-of-00004.safetensors",
710
+ "visual.blocks.8.mlp.fc1.weight": "model-00001-of-00004.safetensors",
711
+ "visual.blocks.8.mlp.fc2.bias": "model-00001-of-00004.safetensors",
712
+ "visual.blocks.8.mlp.fc2.weight": "model-00001-of-00004.safetensors",
713
+ "visual.blocks.8.norm1.bias": "model-00001-of-00004.safetensors",
714
+ "visual.blocks.8.norm1.weight": "model-00001-of-00004.safetensors",
715
+ "visual.blocks.8.norm2.bias": "model-00001-of-00004.safetensors",
716
+ "visual.blocks.8.norm2.weight": "model-00001-of-00004.safetensors",
717
+ "visual.blocks.9.attn.proj.bias": "model-00001-of-00004.safetensors",
718
+ "visual.blocks.9.attn.proj.weight": "model-00001-of-00004.safetensors",
719
+ "visual.blocks.9.attn.qkv.bias": "model-00001-of-00004.safetensors",
720
+ "visual.blocks.9.attn.qkv.weight": "model-00001-of-00004.safetensors",
721
+ "visual.blocks.9.mlp.fc1.bias": "model-00001-of-00004.safetensors",
722
+ "visual.blocks.9.mlp.fc1.weight": "model-00001-of-00004.safetensors",
723
+ "visual.blocks.9.mlp.fc2.bias": "model-00001-of-00004.safetensors",
724
+ "visual.blocks.9.mlp.fc2.weight": "model-00001-of-00004.safetensors",
725
+ "visual.blocks.9.norm1.bias": "model-00001-of-00004.safetensors",
726
+ "visual.blocks.9.norm1.weight": "model-00001-of-00004.safetensors",
727
+ "visual.blocks.9.norm2.bias": "model-00001-of-00004.safetensors",
728
+ "visual.blocks.9.norm2.weight": "model-00001-of-00004.safetensors",
729
+ "visual.merger.ln_q.bias": "model-00001-of-00004.safetensors",
730
+ "visual.merger.ln_q.weight": "model-00001-of-00004.safetensors",
731
+ "visual.merger.mlp.0.bias": "model-00001-of-00004.safetensors",
732
+ "visual.merger.mlp.0.weight": "model-00001-of-00004.safetensors",
733
+ "visual.merger.mlp.2.bias": "model-00001-of-00004.safetensors",
734
+ "visual.merger.mlp.2.weight": "model-00001-of-00004.safetensors",
735
+ "visual.patch_embed.proj.weight": "model-00001-of-00004.safetensors"
736
+ }
737
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "min_pixels": 3136,
3
+ "max_pixels": 12845056,
4
+ "patch_size": 14,
5
+ "temporal_patch_size": 2,
6
+ "merge_size": 2,
7
+ "image_mean": [
8
+ 0.48145466,
9
+ 0.4578275,
10
+ 0.40821073
11
+ ],
12
+ "image_std": [
13
+ 0.26862954,
14
+ 0.26130258,
15
+ 0.27577711
16
+ ],
17
+ "image_processor_type": "Qwen2VLImageProcessor",
18
+ "processor_class": "Qwen2VLProcessor"
19
+ }
tg04_splash.jpg ADDED
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,129 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "151643": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "151644": {
13
+ "content": "<|im_start|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "151645": {
21
+ "content": "<|im_end|>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "151646": {
29
+ "content": "<|object_ref_start|>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "151647": {
37
+ "content": "<|object_ref_end|>",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "151648": {
45
+ "content": "<|box_start|>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "151649": {
53
+ "content": "<|box_end|>",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "151650": {
61
+ "content": "<|quad_start|>",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "151651": {
69
+ "content": "<|quad_end|>",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "151652": {
77
+ "content": "<|vision_start|>",
78
+ "lstrip": false,
79
+ "normalized": false,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": true
83
+ },
84
+ "151653": {
85
+ "content": "<|vision_end|>",
86
+ "lstrip": false,
87
+ "normalized": false,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": true
91
+ },
92
+ "151654": {
93
+ "content": "<|vision_pad|>",
94
+ "lstrip": false,
95
+ "normalized": false,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": true
99
+ },
100
+ "151655": {
101
+ "content": "<|image_pad|>",
102
+ "lstrip": false,
103
+ "normalized": false,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": true
107
+ },
108
+ "151656": {
109
+ "content": "<|video_pad|>",
110
+ "lstrip": false,
111
+ "normalized": false,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": true
115
+ }
116
+ },
117
+ "additional_special_tokens": ["<|im_start|>", "<|im_end|>", "<|object_ref_start|>","<|object_ref_end|>","<|box_start|>","<|box_end|>","<|quad_start|>","<|quad_end|>","<|vision_start|>","<|vision_end|>","<|vision_pad|>","<|image_pad|>","<|video_pad|>"],
118
+ "bos_token": null,
119
+ "chat_template": "{% set image_count = namespace(value=0) %}{% set video_count = namespace(value=0) %}{% for message in messages %}{% if loop.first and message['role'] != 'system' %}<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n{% endif %}<|im_start|>{{ message['role'] }}\n{% if message['content'] is string %}{{ message['content'] }}<|im_end|>\n{% else %}{% for content in message['content'] %}{% if content['type'] == 'image' or 'image' in content or 'image_url' in content %}{% set image_count.value = image_count.value + 1 %}{% if add_vision_id %}Picture {{ image_count.value }}: {% endif %}<|vision_start|><|image_pad|><|vision_end|>{% elif content['type'] == 'video' or 'video' in content %}{% set video_count.value = video_count.value + 1 %}{% if add_vision_id %}Video {{ video_count.value }}: {% endif %}<|vision_start|><|video_pad|><|vision_end|>{% elif 'text' in content %}{{ content['text'] }}{% endif %}{% endfor %}<|im_end|>\n{% endif %}{% endfor %}{% if add_generation_prompt %}<|im_start|>assistant\n{% endif %}",
120
+ "clean_up_tokenization_spaces": false,
121
+ "eos_token": "<|im_end|>",
122
+ "padding_side": "left",
123
+ "errors": "replace",
124
+ "model_max_length": 32768,
125
+ "pad_token": "<|endoftext|>",
126
+ "split_special_tokens": false,
127
+ "tokenizer_class": "Qwen2Tokenizer",
128
+ "unk_token": null
129
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff