final checkpoint and integration - AWQ quantization

README.md

@@ -1,3 +1,32 @@
 ---
 license: apache-2.0
+datasets:
+- HuggingFaceM4/OBELICS
+- laion/laion-coco
+- wikipedia
+- facebook/pmd
+- pixparse/idl-wds
+- pixparse/pdfa-eng-wds
+- wendlerc/RenderedText
+- HuggingFaceM4/the_cauldron
+- teknium/OpenHermes-2.5
+- GAIR/lima
+- databricks/databricks-dolly-15k
+- meta-math/MetaMathQA
+- TIGER-Lab/MathInstruct
+- microsoft/orca-math-word-problems-200k
+- camel-ai/math
+- AtlasUnified/atlas-math-sets
+- tiedong/goat
+language:
+- en
+tags:
+- multimodal
+- vision
+- image-text-to-text
+- quantized
+- 4-bit
+- AWQ
 ---
+
+4-bit AWQ-quantized version of [HuggingFaceM4/idefics2-8b](https://huggingface.co/HuggingFaceM4/idefics2-8b). Refer to the original model's card for more information (including inference snippet).

added_tokens.json

@@ -0,0 +1,5 @@
+{
+  "<end_of_utterance>": 32002,
+  "<fake_token_around_image>": 32000,
+  "<image>": 32001
+}

config.json

@@ -0,0 +1,38 @@
+{
+  "architectures": [
+    "Idefics2ForConditionalGeneration"
+  ],
+  "image_token_id": 32001,
+  "model_type": "idefics2",
+  "perceiver_config": {
+    "model_type": "idefics2"
+  },
+  "quantization_config": {
+    "bits": 4,
+    "group_size": 128,
+    "modules_to_not_convert": ["model.vision_model", "model.modality_projection", "model.perceiver_resampler"],
+    "quant_method": "awq",
+    "version": "gemm",
+    "zero_point": true
+  },
+  "text_config": {
+    "max_position_embeddings": 32768,
+    "model_type": "mistral",
+    "pad_token_id": 0,
+    "rms_norm_eps": 1e-05,
+    "vocab_size": 32003
+  },
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.40.0.dev0",
+  "use_cache": true,
+  "vision_config": {
+    "hidden_size": 1152,
+    "image_size": 980,
+    "intermediate_size": 4304,
+    "model_type": "idefics2",
+    "num_attention_heads": 16,
+    "num_hidden_layers": 27,
+    "patch_size": 14
+  }
+}

generation_config.json

@@ -0,0 +1,18 @@
+{
+  "_from_model_config": true,
+  "bad_words_ids": [
+    [
+      32000
+    ],
+    [
+      32001
+    ]
+  ],
+  "bos_token_id": 1,
+  "eos_token_id": [
+    2,
+    32002
+  ],
+  "pad_token_id": 0,
+  "transformers_version": "4.39.0.dev0"
+}

model-00001-of-00002.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5ccb36242332fc5a7e1cabee75b5370a74ce4e69f8dd1d03bbd464efd8d0273
+size 4985996400

model-00002-of-00002.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f412f25f2a50953c74169d15c092d8f80dfba11be466b112e70056a4cbd59198
+size 1487025616

preprocessor_config.json

@@ -0,0 +1,26 @@
+{
+  "do_convert_rgb": true,
+  "do_image_splitting": true,
+  "do_normalize": true,
+  "do_pad": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "Idefics2ImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "processor_class": "Idefics2Processor",
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "longest_edge": 980,
+    "shortest_edge": 378
+  }
+}

processor_config.json

@@ -0,0 +1,4 @@
+{
+  "image_seq_len": 64,
+  "processor_class": "Idefics2Processor"
+}

special_tokens_map.json

@@ -0,0 +1,53 @@
+{
+  "additional_special_tokens": [
+    {
+      "content": "<fake_token_around_image>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false
+    },
+    {
+      "content": "<image>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false
+    },
+    {
+      "content": "<end_of_utterance>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false
+    }
+  ],
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dadfd56d766715c61d2ef780a525ab43b8e6da4de6865bda3d95fdef5e134055
+size 493443

tokenizer_config.json

@@ -0,0 +1,71 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32000": {
+      "content": "<fake_token_around_image>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32001": {
+      "content": "<image>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32002": {
+      "content": "<end_of_utterance>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<fake_token_around_image>",
+    "<image>",
+    "<end_of_utterance>"
+  ],
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "legacy": false,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<unk>",
+  "processor_class": "Idefics2Processor",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": true
+}