add model

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text

chat_template.json

@@ -0,0 +1,3 @@
+{
+  "chat_template": "{{- bos_token }}\n{%- if custom_tools is defined %}\n    {%- set tools = custom_tools %}\n{%- endif %}\n{%- if not tools_in_user_message is defined %}\n    {%- set tools_in_user_message = true %}\n{%- endif %}\n{%- if not date_string is defined %}\n    {%- if strftime_now is defined %}\n        {%- set date_string = strftime_now(\"%d %b %Y\") %}\n    {%- else %}\n        {%- set date_string = \"26 Jul 2024\" %}\n    {%- endif %}\n{%- endif %}\n{%- if not tools is defined %}\n    {%- set tools = none %}\n{%- endif %}\n\n{#- This block extracts the system message, so we can slot it into the right place. #}\n{%- if messages[0]['role'] == 'system' %}    \n    {%- if messages[0]['content'] is string %}\n        {%- set system_message = messages[0]['content']|trim %}\n    {%- else %}\n        {#- FIXME: The processor requires an array, always. #}\n        {%- set system_message = messages[0]['content'][0]['text']|trim %}\n    {%- endif %}\n    {%- set messages = messages[1:] %}\n    {%- set user_supplied_system_message = true %}\n{%- else %}\n    {%- set system_message = \"\" %}\n    {%- set user_supplied_system_message = false %}\n{%- endif %}\n\n{#- System message if the user supplied one #}\n{%- if user_supplied_system_message %}\n    {{- \"<|header_start|>system<|header_end|>\\n\\n\" }}\n    {%- if tools is not none %}\n        {{- \"Environment: ipython\\n\" }}\n    {%- endif %}\n    {%- if tools is not none and not tools_in_user_message %}\n        {{- \"You have access to the following functions. To call a function, please respond with JSON for a function call.\" }}\n        {{- 'Respond in the format {\"name\": function name, \"parameters\": dictionary of argument name and its value}.' }}\n        {{- \"Do not use variables.\\n\\n\" }}\n        {%- for t in tools %}\n            {{- t | tojson(indent=4) }}\n            {{- \"\\n\\n\" }}\n        {%- endfor %}\n    {%- endif %}\n    {{- system_message }}\n    {{- \"<|eot|>\" }}\n{%- endif %}\n\n{#- Custom tools are passed in a user message with some extra guidance #}\n{%- if tools_in_user_message and not tools is none %}\n    {#- Extract the first user message so we can plug it in here #}\n    {%- if messages | length != 0 %}\n        {%- set first_user_message = messages[0]['content']|trim %}\n        {%- set messages = messages[1:] %}\n    {%- else %}\n        {{- raise_exception(\"Cannot put tools in the first user message when there's no first user message!\") }}\n{%- endif %}\n    {{- '<|header_start|>user<|header_end|>\\n\\n' -}}\n    {{- \"Given the following functions, please respond with a JSON for a function call \" }}\n    {{- \"with its proper arguments that best answers the given prompt.\\n\\n\" }}\n    {{- 'Respond in the format {\"name\": function name, \"parameters\": dictionary of argument name and its value}.' }}\n    {{- \"Do not use variables.\\n\\n\" }}\n    {%- for t in tools %}\n        {{- t | tojson(indent=4) }}\n        {{- \"\\n\\n\" }}\n    {%- endfor %}\n    {{- first_user_message + \"<|eot|>\"}}\n{%- endif %}\n\n{%- for message in messages %}\n    {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}\n    {{- '<|header_start|>' + message['role'] + '<|header_end|>\\n\\n' }}\n        {%- if message['content'] is string %}\n            {{- message['content'] }}\n        {%- else %}\n            {%- for content in message['content'] %}\n                {%- if content['type'] == 'image' %}\n                    {{- '<|image|>' }}\n                {%- elif content['type'] == 'text' %}\n                    {{- content['text'] }}\n                {%- endif %}\n            {%- endfor %}\n        {%- endif %}\n        {{- \"<|eot|>\" }}\n    {%- elif 'tool_calls' in message and message.tool_calls|length > 0 %}\n       {{- '<|header_start|>assistant<|header_end|>\\n\\n' -}}\n       {{- '<|python_start|>' }}\n        {%- if message['content'] is string %}\n            {{- message['content'] }}\n        {%- else %}\n            {%- for content in message['content'] %}\n                {%- if content['type'] == 'image' %}\n                    {{- '<|image|>' }}\n                {%- elif content['type'] == 'text' %}\n                    {{- content['text'] }}\n                {%- endif %}\n            {%- endfor %}\n        {%- endif %}\n       {{- '<|python_end|>' }}\n        {%- for tool_call in message.tool_calls %}\n           {{- '{\"name\": \"' + tool_call.function.name + '\", ' }}\n           {{- '\"parameters\": ' }}\n           {{- tool_call.function.arguments | tojson }}\n           {{- \"}\" }}\n        {%- endfor %}\n       {{- \"<|eot|>\" }}\n    {%- elif message.role == \"tool\" or message.role == \"ipython\" %}\n        {{- \"<|header_start|>ipython<|header_end|>\\n\\n\" }}\n        {%- if message.content is mapping or message.content is iterable %}\n            {{- message.content | tojson }}\n        {%- else %}\n            {{- message.content }}\n        {%- endif %}\n        {{- \"<|eot|>\" }}\n    {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n    {{- '<|header_start|>assistant<|header_end|>\\n\\n' }}\n{%- endif %}\n"
+}

config.json

@@ -0,0 +1,662 @@
+{
+  "architectures": [
+    "Llama4ForConditionalGeneration"
+  ],
+  "boi_token_index": 200080,
+  "eoi_token_index": 200081,
+  "image_token_index": 200092,
+  "model_type": "llama4",
+  "quantization_config": {
+    "config_groups": {
+      "group_0": {
+        "input_activations": null,
+        "output_activations": null,
+        "targets": [
+          "Linear"
+        ],
+        "weights": {
+          "actorder": "weight",
+          "block_structure": null,
+          "dynamic": false,
+          "group_size": 128,
+          "num_bits": 4,
+          "observer": "minmax",
+          "observer_kwargs": {},
+          "strategy": "group",
+          "symmetric": true,
+          "type": "int"
+        }
+      }
+    },
+    "format": "pack-quantized",
+    "global_compression_ratio": null,
+    "ignore": [
+      "vision_model.patch_embedding.linear",
+      "vision_model.model.layers.0.self_attn.q_proj",
+      "vision_model.model.layers.0.self_attn.k_proj",
+      "vision_model.model.layers.0.self_attn.v_proj",
+      "vision_model.model.layers.0.self_attn.o_proj",
+      "vision_model.model.layers.0.mlp.fc1",
+      "vision_model.model.layers.0.mlp.fc2",
+      "vision_model.model.layers.1.self_attn.q_proj",
+      "vision_model.model.layers.1.self_attn.k_proj",
+      "vision_model.model.layers.1.self_attn.v_proj",
+      "vision_model.model.layers.1.self_attn.o_proj",
+      "vision_model.model.layers.1.mlp.fc1",
+      "vision_model.model.layers.1.mlp.fc2",
+      "vision_model.model.layers.2.self_attn.q_proj",
+      "vision_model.model.layers.2.self_attn.k_proj",
+      "vision_model.model.layers.2.self_attn.v_proj",
+      "vision_model.model.layers.2.self_attn.o_proj",
+      "vision_model.model.layers.2.mlp.fc1",
+      "vision_model.model.layers.2.mlp.fc2",
+      "vision_model.model.layers.3.self_attn.q_proj",
+      "vision_model.model.layers.3.self_attn.k_proj",
+      "vision_model.model.layers.3.self_attn.v_proj",
+      "vision_model.model.layers.3.self_attn.o_proj",
+      "vision_model.model.layers.3.mlp.fc1",
+      "vision_model.model.layers.3.mlp.fc2",
+      "vision_model.model.layers.4.self_attn.q_proj",
+      "vision_model.model.layers.4.self_attn.k_proj",
+      "vision_model.model.layers.4.self_attn.v_proj",
+      "vision_model.model.layers.4.self_attn.o_proj",
+      "vision_model.model.layers.4.mlp.fc1",
+      "vision_model.model.layers.4.mlp.fc2",
+      "vision_model.model.layers.5.self_attn.q_proj",
+      "vision_model.model.layers.5.self_attn.k_proj",
+      "vision_model.model.layers.5.self_attn.v_proj",
+      "vision_model.model.layers.5.self_attn.o_proj",
+      "vision_model.model.layers.5.mlp.fc1",
+      "vision_model.model.layers.5.mlp.fc2",
+      "vision_model.model.layers.6.self_attn.q_proj",
+      "vision_model.model.layers.6.self_attn.k_proj",
+      "vision_model.model.layers.6.self_attn.v_proj",
+      "vision_model.model.layers.6.self_attn.o_proj",
+      "vision_model.model.layers.6.mlp.fc1",
+      "vision_model.model.layers.6.mlp.fc2",
+      "vision_model.model.layers.7.self_attn.q_proj",
+      "vision_model.model.layers.7.self_attn.k_proj",
+      "vision_model.model.layers.7.self_attn.v_proj",
+      "vision_model.model.layers.7.self_attn.o_proj",
+      "vision_model.model.layers.7.mlp.fc1",
+      "vision_model.model.layers.7.mlp.fc2",
+      "vision_model.model.layers.8.self_attn.q_proj",
+      "vision_model.model.layers.8.self_attn.k_proj",
+      "vision_model.model.layers.8.self_attn.v_proj",
+      "vision_model.model.layers.8.self_attn.o_proj",
+      "vision_model.model.layers.8.mlp.fc1",
+      "vision_model.model.layers.8.mlp.fc2",
+      "vision_model.model.layers.9.self_attn.q_proj",
+      "vision_model.model.layers.9.self_attn.k_proj",
+      "vision_model.model.layers.9.self_attn.v_proj",
+      "vision_model.model.layers.9.self_attn.o_proj",
+      "vision_model.model.layers.9.mlp.fc1",
+      "vision_model.model.layers.9.mlp.fc2",
+      "vision_model.model.layers.10.self_attn.q_proj",
+      "vision_model.model.layers.10.self_attn.k_proj",
+      "vision_model.model.layers.10.self_attn.v_proj",
+      "vision_model.model.layers.10.self_attn.o_proj",
+      "vision_model.model.layers.10.mlp.fc1",
+      "vision_model.model.layers.10.mlp.fc2",
+      "vision_model.model.layers.11.self_attn.q_proj",
+      "vision_model.model.layers.11.self_attn.k_proj",
+      "vision_model.model.layers.11.self_attn.v_proj",
+      "vision_model.model.layers.11.self_attn.o_proj",
+      "vision_model.model.layers.11.mlp.fc1",
+      "vision_model.model.layers.11.mlp.fc2",
+      "vision_model.model.layers.12.self_attn.q_proj",
+      "vision_model.model.layers.12.self_attn.k_proj",
+      "vision_model.model.layers.12.self_attn.v_proj",
+      "vision_model.model.layers.12.self_attn.o_proj",
+      "vision_model.model.layers.12.mlp.fc1",
+      "vision_model.model.layers.12.mlp.fc2",
+      "vision_model.model.layers.13.self_attn.q_proj",
+      "vision_model.model.layers.13.self_attn.k_proj",
+      "vision_model.model.layers.13.self_attn.v_proj",
+      "vision_model.model.layers.13.self_attn.o_proj",
+      "vision_model.model.layers.13.mlp.fc1",
+      "vision_model.model.layers.13.mlp.fc2",
+      "vision_model.model.layers.14.self_attn.q_proj",
+      "vision_model.model.layers.14.self_attn.k_proj",
+      "vision_model.model.layers.14.self_attn.v_proj",
+      "vision_model.model.layers.14.self_attn.o_proj",
+      "vision_model.model.layers.14.mlp.fc1",
+      "vision_model.model.layers.14.mlp.fc2",
+      "vision_model.model.layers.15.self_attn.q_proj",
+      "vision_model.model.layers.15.self_attn.k_proj",
+      "vision_model.model.layers.15.self_attn.v_proj",
+      "vision_model.model.layers.15.self_attn.o_proj",
+      "vision_model.model.layers.15.mlp.fc1",
+      "vision_model.model.layers.15.mlp.fc2",
+      "vision_model.model.layers.16.self_attn.q_proj",
+      "vision_model.model.layers.16.self_attn.k_proj",
+      "vision_model.model.layers.16.self_attn.v_proj",
+      "vision_model.model.layers.16.self_attn.o_proj",
+      "vision_model.model.layers.16.mlp.fc1",
+      "vision_model.model.layers.16.mlp.fc2",
+      "vision_model.model.layers.17.self_attn.q_proj",
+      "vision_model.model.layers.17.self_attn.k_proj",
+      "vision_model.model.layers.17.self_attn.v_proj",
+      "vision_model.model.layers.17.self_attn.o_proj",
+      "vision_model.model.layers.17.mlp.fc1",
+      "vision_model.model.layers.17.mlp.fc2",
+      "vision_model.model.layers.18.self_attn.q_proj",
+      "vision_model.model.layers.18.self_attn.k_proj",
+      "vision_model.model.layers.18.self_attn.v_proj",
+      "vision_model.model.layers.18.self_attn.o_proj",
+      "vision_model.model.layers.18.mlp.fc1",
+      "vision_model.model.layers.18.mlp.fc2",
+      "vision_model.model.layers.19.self_attn.q_proj",
+      "vision_model.model.layers.19.self_attn.k_proj",
+      "vision_model.model.layers.19.self_attn.v_proj",
+      "vision_model.model.layers.19.self_attn.o_proj",
+      "vision_model.model.layers.19.mlp.fc1",
+      "vision_model.model.layers.19.mlp.fc2",
+      "vision_model.model.layers.20.self_attn.q_proj",
+      "vision_model.model.layers.20.self_attn.k_proj",
+      "vision_model.model.layers.20.self_attn.v_proj",
+      "vision_model.model.layers.20.self_attn.o_proj",
+      "vision_model.model.layers.20.mlp.fc1",
+      "vision_model.model.layers.20.mlp.fc2",
+      "vision_model.model.layers.21.self_attn.q_proj",
+      "vision_model.model.layers.21.self_attn.k_proj",
+      "vision_model.model.layers.21.self_attn.v_proj",
+      "vision_model.model.layers.21.self_attn.o_proj",
+      "vision_model.model.layers.21.mlp.fc1",
+      "vision_model.model.layers.21.mlp.fc2",
+      "vision_model.model.layers.22.self_attn.q_proj",
+      "vision_model.model.layers.22.self_attn.k_proj",
+      "vision_model.model.layers.22.self_attn.v_proj",
+      "vision_model.model.layers.22.self_attn.o_proj",
+      "vision_model.model.layers.22.mlp.fc1",
+      "vision_model.model.layers.22.mlp.fc2",
+      "vision_model.model.layers.23.self_attn.q_proj",
+      "vision_model.model.layers.23.self_attn.k_proj",
+      "vision_model.model.layers.23.self_attn.v_proj",
+      "vision_model.model.layers.23.self_attn.o_proj",
+      "vision_model.model.layers.23.mlp.fc1",
+      "vision_model.model.layers.23.mlp.fc2",
+      "vision_model.model.layers.24.self_attn.q_proj",
+      "vision_model.model.layers.24.self_attn.k_proj",
+      "vision_model.model.layers.24.self_attn.v_proj",
+      "vision_model.model.layers.24.self_attn.o_proj",
+      "vision_model.model.layers.24.mlp.fc1",
+      "vision_model.model.layers.24.mlp.fc2",
+      "vision_model.model.layers.25.self_attn.q_proj",
+      "vision_model.model.layers.25.self_attn.k_proj",
+      "vision_model.model.layers.25.self_attn.v_proj",
+      "vision_model.model.layers.25.self_attn.o_proj",
+      "vision_model.model.layers.25.mlp.fc1",
+      "vision_model.model.layers.25.mlp.fc2",
+      "vision_model.model.layers.26.self_attn.q_proj",
+      "vision_model.model.layers.26.self_attn.k_proj",
+      "vision_model.model.layers.26.self_attn.v_proj",
+      "vision_model.model.layers.26.self_attn.o_proj",
+      "vision_model.model.layers.26.mlp.fc1",
+      "vision_model.model.layers.26.mlp.fc2",
+      "vision_model.model.layers.27.self_attn.q_proj",
+      "vision_model.model.layers.27.self_attn.k_proj",
+      "vision_model.model.layers.27.self_attn.v_proj",
+      "vision_model.model.layers.27.self_attn.o_proj",
+      "vision_model.model.layers.27.mlp.fc1",
+      "vision_model.model.layers.27.mlp.fc2",
+      "vision_model.model.layers.28.self_attn.q_proj",
+      "vision_model.model.layers.28.self_attn.k_proj",
+      "vision_model.model.layers.28.self_attn.v_proj",
+      "vision_model.model.layers.28.self_attn.o_proj",
+      "vision_model.model.layers.28.mlp.fc1",
+      "vision_model.model.layers.28.mlp.fc2",
+      "vision_model.model.layers.29.self_attn.q_proj",
+      "vision_model.model.layers.29.self_attn.k_proj",
+      "vision_model.model.layers.29.self_attn.v_proj",
+      "vision_model.model.layers.29.self_attn.o_proj",
+      "vision_model.model.layers.29.mlp.fc1",
+      "vision_model.model.layers.29.mlp.fc2",
+      "vision_model.model.layers.30.self_attn.q_proj",
+      "vision_model.model.layers.30.self_attn.k_proj",
+      "vision_model.model.layers.30.self_attn.v_proj",
+      "vision_model.model.layers.30.self_attn.o_proj",
+      "vision_model.model.layers.30.mlp.fc1",
+      "vision_model.model.layers.30.mlp.fc2",
+      "vision_model.model.layers.31.self_attn.q_proj",
+      "vision_model.model.layers.31.self_attn.k_proj",
+      "vision_model.model.layers.31.self_attn.v_proj",
+      "vision_model.model.layers.31.self_attn.o_proj",
+      "vision_model.model.layers.31.mlp.fc1",
+      "vision_model.model.layers.31.mlp.fc2",
+      "vision_model.model.layers.32.self_attn.q_proj",
+      "vision_model.model.layers.32.self_attn.k_proj",
+      "vision_model.model.layers.32.self_attn.v_proj",
+      "vision_model.model.layers.32.self_attn.o_proj",
+      "vision_model.model.layers.32.mlp.fc1",
+      "vision_model.model.layers.32.mlp.fc2",
+      "vision_model.model.layers.33.self_attn.q_proj",
+      "vision_model.model.layers.33.self_attn.k_proj",
+      "vision_model.model.layers.33.self_attn.v_proj",
+      "vision_model.model.layers.33.self_attn.o_proj",
+      "vision_model.model.layers.33.mlp.fc1",
+      "vision_model.model.layers.33.mlp.fc2",
+      "vision_model.vision_adapter.mlp.fc1",
+      "vision_model.vision_adapter.mlp.fc2",
+      "multi_modal_projector.linear_1",
+      "language_model.model.layers.0.self_attn.q_proj",
+      "language_model.model.layers.0.self_attn.k_proj",
+      "language_model.model.layers.0.self_attn.v_proj",
+      "language_model.model.layers.0.self_attn.o_proj",
+      "language_model.model.layers.0.feed_forward.router",
+      "language_model.model.layers.1.self_attn.q_proj",
+      "language_model.model.layers.1.self_attn.k_proj",
+      "language_model.model.layers.1.self_attn.v_proj",
+      "language_model.model.layers.1.self_attn.o_proj",
+      "language_model.model.layers.1.feed_forward.router",
+      "language_model.model.layers.2.self_attn.q_proj",
+      "language_model.model.layers.2.self_attn.k_proj",
+      "language_model.model.layers.2.self_attn.v_proj",
+      "language_model.model.layers.2.self_attn.o_proj",
+      "language_model.model.layers.2.feed_forward.router",
+      "language_model.model.layers.3.self_attn.q_proj",
+      "language_model.model.layers.3.self_attn.k_proj",
+      "language_model.model.layers.3.self_attn.v_proj",
+      "language_model.model.layers.3.self_attn.o_proj",
+      "language_model.model.layers.3.feed_forward.router",
+      "language_model.model.layers.4.self_attn.q_proj",
+      "language_model.model.layers.4.self_attn.k_proj",
+      "language_model.model.layers.4.self_attn.v_proj",
+      "language_model.model.layers.4.self_attn.o_proj",
+      "language_model.model.layers.4.feed_forward.router",
+      "language_model.model.layers.5.self_attn.q_proj",
+      "language_model.model.layers.5.self_attn.k_proj",
+      "language_model.model.layers.5.self_attn.v_proj",
+      "language_model.model.layers.5.self_attn.o_proj",
+      "language_model.model.layers.5.feed_forward.router",
+      "language_model.model.layers.6.self_attn.q_proj",
+      "language_model.model.layers.6.self_attn.k_proj",
+      "language_model.model.layers.6.self_attn.v_proj",
+      "language_model.model.layers.6.self_attn.o_proj",
+      "language_model.model.layers.6.feed_forward.router",
+      "language_model.model.layers.7.self_attn.q_proj",
+      "language_model.model.layers.7.self_attn.k_proj",
+      "language_model.model.layers.7.self_attn.v_proj",
+      "language_model.model.layers.7.self_attn.o_proj",
+      "language_model.model.layers.7.feed_forward.router",
+      "language_model.model.layers.8.self_attn.q_proj",
+      "language_model.model.layers.8.self_attn.k_proj",
+      "language_model.model.layers.8.self_attn.v_proj",
+      "language_model.model.layers.8.self_attn.o_proj",
+      "language_model.model.layers.8.feed_forward.router",
+      "language_model.model.layers.9.self_attn.q_proj",
+      "language_model.model.layers.9.self_attn.k_proj",
+      "language_model.model.layers.9.self_attn.v_proj",
+      "language_model.model.layers.9.self_attn.o_proj",
+      "language_model.model.layers.9.feed_forward.router",
+      "language_model.model.layers.10.self_attn.q_proj",
+      "language_model.model.layers.10.self_attn.k_proj",
+      "language_model.model.layers.10.self_attn.v_proj",
+      "language_model.model.layers.10.self_attn.o_proj",
+      "language_model.model.layers.10.feed_forward.router",
+      "language_model.model.layers.11.self_attn.q_proj",
+      "language_model.model.layers.11.self_attn.k_proj",
+      "language_model.model.layers.11.self_attn.v_proj",
+      "language_model.model.layers.11.self_attn.o_proj",
+      "language_model.model.layers.11.feed_forward.router",
+      "language_model.model.layers.12.self_attn.q_proj",
+      "language_model.model.layers.12.self_attn.k_proj",
+      "language_model.model.layers.12.self_attn.v_proj",
+      "language_model.model.layers.12.self_attn.o_proj",
+      "language_model.model.layers.12.feed_forward.router",
+      "language_model.model.layers.13.self_attn.q_proj",
+      "language_model.model.layers.13.self_attn.k_proj",
+      "language_model.model.layers.13.self_attn.v_proj",
+      "language_model.model.layers.13.self_attn.o_proj",
+      "language_model.model.layers.13.feed_forward.router",
+      "language_model.model.layers.14.self_attn.q_proj",
+      "language_model.model.layers.14.self_attn.k_proj",
+      "language_model.model.layers.14.self_attn.v_proj",
+      "language_model.model.layers.14.self_attn.o_proj",
+      "language_model.model.layers.14.feed_forward.router",
+      "language_model.model.layers.15.self_attn.q_proj",
+      "language_model.model.layers.15.self_attn.k_proj",
+      "language_model.model.layers.15.self_attn.v_proj",
+      "language_model.model.layers.15.self_attn.o_proj",
+      "language_model.model.layers.15.feed_forward.router",
+      "language_model.model.layers.16.self_attn.q_proj",
+      "language_model.model.layers.16.self_attn.k_proj",
+      "language_model.model.layers.16.self_attn.v_proj",
+      "language_model.model.layers.16.self_attn.o_proj",
+      "language_model.model.layers.16.feed_forward.router",
+      "language_model.model.layers.17.self_attn.q_proj",
+      "language_model.model.layers.17.self_attn.k_proj",
+      "language_model.model.layers.17.self_attn.v_proj",
+      "language_model.model.layers.17.self_attn.o_proj",
+      "language_model.model.layers.17.feed_forward.router",
+      "language_model.model.layers.18.self_attn.q_proj",
+      "language_model.model.layers.18.self_attn.k_proj",
+      "language_model.model.layers.18.self_attn.v_proj",
+      "language_model.model.layers.18.self_attn.o_proj",
+      "language_model.model.layers.18.feed_forward.router",
+      "language_model.model.layers.19.self_attn.q_proj",
+      "language_model.model.layers.19.self_attn.k_proj",
+      "language_model.model.layers.19.self_attn.v_proj",
+      "language_model.model.layers.19.self_attn.o_proj",
+      "language_model.model.layers.19.feed_forward.router",
+      "language_model.model.layers.20.self_attn.q_proj",
+      "language_model.model.layers.20.self_attn.k_proj",
+      "language_model.model.layers.20.self_attn.v_proj",
+      "language_model.model.layers.20.self_attn.o_proj",
+      "language_model.model.layers.20.feed_forward.router",
+      "language_model.model.layers.21.self_attn.q_proj",
+      "language_model.model.layers.21.self_attn.k_proj",
+      "language_model.model.layers.21.self_attn.v_proj",
+      "language_model.model.layers.21.self_attn.o_proj",
+      "language_model.model.layers.21.feed_forward.router",
+      "language_model.model.layers.22.self_attn.q_proj",
+      "language_model.model.layers.22.self_attn.k_proj",
+      "language_model.model.layers.22.self_attn.v_proj",
+      "language_model.model.layers.22.self_attn.o_proj",
+      "language_model.model.layers.22.feed_forward.router",
+      "language_model.model.layers.23.self_attn.q_proj",
+      "language_model.model.layers.23.self_attn.k_proj",
+      "language_model.model.layers.23.self_attn.v_proj",
+      "language_model.model.layers.23.self_attn.o_proj",
+      "language_model.model.layers.23.feed_forward.router",
+      "language_model.model.layers.24.self_attn.q_proj",
+      "language_model.model.layers.24.self_attn.k_proj",
+      "language_model.model.layers.24.self_attn.v_proj",
+      "language_model.model.layers.24.self_attn.o_proj",
+      "language_model.model.layers.24.feed_forward.router",
+      "language_model.model.layers.25.self_attn.q_proj",
+      "language_model.model.layers.25.self_attn.k_proj",
+      "language_model.model.layers.25.self_attn.v_proj",
+      "language_model.model.layers.25.self_attn.o_proj",
+      "language_model.model.layers.25.feed_forward.router",
+      "language_model.model.layers.26.self_attn.q_proj",
+      "language_model.model.layers.26.self_attn.k_proj",
+      "language_model.model.layers.26.self_attn.v_proj",
+      "language_model.model.layers.26.self_attn.o_proj",
+      "language_model.model.layers.26.feed_forward.router",
+      "language_model.model.layers.27.self_attn.q_proj",
+      "language_model.model.layers.27.self_attn.k_proj",
+      "language_model.model.layers.27.self_attn.v_proj",
+      "language_model.model.layers.27.self_attn.o_proj",
+      "language_model.model.layers.27.feed_forward.router",
+      "language_model.model.layers.28.self_attn.q_proj",
+      "language_model.model.layers.28.self_attn.k_proj",
+      "language_model.model.layers.28.self_attn.v_proj",
+      "language_model.model.layers.28.self_attn.o_proj",
+      "language_model.model.layers.28.feed_forward.router",
+      "language_model.model.layers.29.self_attn.q_proj",
+      "language_model.model.layers.29.self_attn.k_proj",
+      "language_model.model.layers.29.self_attn.v_proj",
+      "language_model.model.layers.29.self_attn.o_proj",
+      "language_model.model.layers.29.feed_forward.router",
+      "language_model.model.layers.30.self_attn.q_proj",
+      "language_model.model.layers.30.self_attn.k_proj",
+      "language_model.model.layers.30.self_attn.v_proj",
+      "language_model.model.layers.30.self_attn.o_proj",
+      "language_model.model.layers.30.feed_forward.router",
+      "language_model.model.layers.31.self_attn.q_proj",
+      "language_model.model.layers.31.self_attn.k_proj",
+      "language_model.model.layers.31.self_attn.v_proj",
+      "language_model.model.layers.31.self_attn.o_proj",
+      "language_model.model.layers.31.feed_forward.router",
+      "language_model.model.layers.32.self_attn.q_proj",
+      "language_model.model.layers.32.self_attn.k_proj",
+      "language_model.model.layers.32.self_attn.v_proj",
+      "language_model.model.layers.32.self_attn.o_proj",
+      "language_model.model.layers.32.feed_forward.router",
+      "language_model.model.layers.33.self_attn.q_proj",
+      "language_model.model.layers.33.self_attn.k_proj",
+      "language_model.model.layers.33.self_attn.v_proj",
+      "language_model.model.layers.33.self_attn.o_proj",
+      "language_model.model.layers.33.feed_forward.router",
+      "language_model.model.layers.34.self_attn.q_proj",
+      "language_model.model.layers.34.self_attn.k_proj",
+      "language_model.model.layers.34.self_attn.v_proj",
+      "language_model.model.layers.34.self_attn.o_proj",
+      "language_model.model.layers.34.feed_forward.router",
+      "language_model.model.layers.35.self_attn.q_proj",
+      "language_model.model.layers.35.self_attn.k_proj",
+      "language_model.model.layers.35.self_attn.v_proj",
+      "language_model.model.layers.35.self_attn.o_proj",
+      "language_model.model.layers.35.feed_forward.router",
+      "language_model.model.layers.36.self_attn.q_proj",
+      "language_model.model.layers.36.self_attn.k_proj",
+      "language_model.model.layers.36.self_attn.v_proj",
+      "language_model.model.layers.36.self_attn.o_proj",
+      "language_model.model.layers.36.feed_forward.router",
+      "language_model.model.layers.37.self_attn.q_proj",
+      "language_model.model.layers.37.self_attn.k_proj",
+      "language_model.model.layers.37.self_attn.v_proj",
+      "language_model.model.layers.37.self_attn.o_proj",
+      "language_model.model.layers.37.feed_forward.router",
+      "language_model.model.layers.38.self_attn.q_proj",
+      "language_model.model.layers.38.self_attn.k_proj",
+      "language_model.model.layers.38.self_attn.v_proj",
+      "language_model.model.layers.38.self_attn.o_proj",
+      "language_model.model.layers.38.feed_forward.router",
+      "language_model.model.layers.39.self_attn.q_proj",
+      "language_model.model.layers.39.self_attn.k_proj",
+      "language_model.model.layers.39.self_attn.v_proj",
+      "language_model.model.layers.39.self_attn.o_proj",
+      "language_model.model.layers.39.feed_forward.router",
+      "language_model.model.layers.40.self_attn.q_proj",
+      "language_model.model.layers.40.self_attn.k_proj",
+      "language_model.model.layers.40.self_attn.v_proj",
+      "language_model.model.layers.40.self_attn.o_proj",
+      "language_model.model.layers.40.feed_forward.router",
+      "language_model.model.layers.41.self_attn.q_proj",
+      "language_model.model.layers.41.self_attn.k_proj",
+      "language_model.model.layers.41.self_attn.v_proj",
+      "language_model.model.layers.41.self_attn.o_proj",
+      "language_model.model.layers.41.feed_forward.router",
+      "language_model.model.layers.42.self_attn.q_proj",
+      "language_model.model.layers.42.self_attn.k_proj",
+      "language_model.model.layers.42.self_attn.v_proj",
+      "language_model.model.layers.42.self_attn.o_proj",
+      "language_model.model.layers.42.feed_forward.router",
+      "language_model.model.layers.43.self_attn.q_proj",
+      "language_model.model.layers.43.self_attn.k_proj",
+      "language_model.model.layers.43.self_attn.v_proj",
+      "language_model.model.layers.43.self_attn.o_proj",
+      "language_model.model.layers.43.feed_forward.router",
+      "language_model.model.layers.44.self_attn.q_proj",
+      "language_model.model.layers.44.self_attn.k_proj",
+      "language_model.model.layers.44.self_attn.v_proj",
+      "language_model.model.layers.44.self_attn.o_proj",
+      "language_model.model.layers.44.feed_forward.router",
+      "language_model.model.layers.45.self_attn.q_proj",
+      "language_model.model.layers.45.self_attn.k_proj",
+      "language_model.model.layers.45.self_attn.v_proj",
+      "language_model.model.layers.45.self_attn.o_proj",
+      "language_model.model.layers.45.feed_forward.router",
+      "language_model.model.layers.46.self_attn.q_proj",
+      "language_model.model.layers.46.self_attn.k_proj",
+      "language_model.model.layers.46.self_attn.v_proj",
+      "language_model.model.layers.46.self_attn.o_proj",
+      "language_model.model.layers.46.feed_forward.router",
+      "language_model.model.layers.47.self_attn.q_proj",
+      "language_model.model.layers.47.self_attn.k_proj",
+      "language_model.model.layers.47.self_attn.v_proj",
+      "language_model.model.layers.47.self_attn.o_proj",
+      "language_model.model.layers.47.feed_forward.router",
+      "language_model.lm_head"
+    ],
+    "kv_cache_scheme": null,
+    "quant_method": "compressed-tensors",
+    "quantization_status": "compressed"
+  },
+  "text_config": {
+    "_attn_implementation_autoset": true,
+    "attention_bias": false,
+    "attention_chunk_size": 8192,
+    "attention_dropout": 0.0,
+    "attn_scale": 0.1,
+    "attn_temperature_tuning": 4,
+    "bos_token_id": 200000,
+    "eos_token_id": [
+      200001,
+      200007,
+      200008
+    ],
+    "floor_scale": 8192,
+    "for_llm_compressor": true,
+    "head_dim": 128,
+    "hidden_act": "silu",
+    "hidden_size": 5120,
+    "initializer_range": 0.02,
+    "interleave_moe_layer_step": 1,
+    "intermediate_size": 8192,
+    "intermediate_size_mlp": 16384,
+    "max_position_embeddings": 10485760,
+    "model_type": "llama4_text",
+    "moe_layers": [
+      0,
+      1,
+      2,
+      3,
+      4,
+      5,
+      6,
+      7,
+      8,
+      9,
+      10,
+      11,
+      12,
+      13,
+      14,
+      15,
+      16,
+      17,
+      18,
+      19,
+      20,
+      21,
+      22,
+      23,
+      24,
+      25,
+      26,
+      27,
+      28,
+      29,
+      30,
+      31,
+      32,
+      33,
+      34,
+      35,
+      36,
+      37,
+      38,
+      39,
+      40,
+      41,
+      42,
+      43,
+      44,
+      45,
+      46,
+      47
+    ],
+    "no_rope_layers": [
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0,
+      1,
+      1,
+      1,
+      0
+    ],
+    "num_attention_heads": 40,
+    "num_experts_per_tok": 1,
+    "num_hidden_layers": 48,
+    "num_key_value_heads": 8,
+    "num_local_experts": 16,
+    "output_router_logits": false,
+    "pad_token_id": 200018,
+    "rms_norm_eps": 1e-05,
+    "rope_scaling": {
+      "factor": 16.0,
+      "high_freq_factor": 1.0,
+      "low_freq_factor": 1.0,
+      "original_max_position_embeddings": 8192,
+      "rope_type": "llama3"
+    },
+    "rope_theta": 500000.0,
+    "router_aux_loss_coef": 0.001,
+    "router_jitter_noise": 0.0,
+    "torch_dtype": "bfloat16",
+    "use_cache": true,
+    "use_qk_norm": true,
+    "vocab_size": 202048
+  },
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.0.dev0",
+  "vision_config": {
+    "_attn_implementation_autoset": true,
+    "attention_dropout": 0.0,
+    "hidden_act": "gelu",
+    "hidden_size": 1408,
+    "image_size": 336,
+    "initializer_range": 0.02,
+    "intermediate_size": 5632,
+    "model_type": "llama4_vision_model",
+    "multi_modal_projector_bias": false,
+    "norm_eps": 1e-05,
+    "num_attention_heads": 16,
+    "num_channels": 3,
+    "num_hidden_layers": 34,
+    "patch_size": 14,
+    "pixel_shuffle_ratio": 0.5,
+    "projector_dropout": 0.0,
+    "projector_input_dim": 4096,
+    "projector_output_dim": 4096,
+    "rope_theta": 10000,
+    "vision_feature_layer": -1,
+    "vision_feature_select_strategy": "default",
+    "vision_output_dim": 4096
+  }
+}

generation_config.json

@@ -0,0 +1,13 @@
+{
+  "bos_token_id": 200000,
+  "do_sample": true,
+  "eos_token_id": [
+    200001,
+    200007,
+    200008
+  ],
+  "pad_token_id": 200018,
+  "temperature": 0.6,
+  "top_p": 0.9,
+  "transformers_version": "4.51.0.dev0"
+}

model-00001-of-00014.safetensors

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

model-00002-of-00014.safetensors

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

model-00003-of-00014.safetensors

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

model-00004-of-00014.safetensors

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

model-00005-of-00014.safetensors

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

model-00006-of-00014.safetensors

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

model-00007-of-00014.safetensors

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

model-00008-of-00014.safetensors

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

model-00009-of-00014.safetensors

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

model-00010-of-00014.safetensors

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

model-00011-of-00014.safetensors

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

model-00012-of-00014.safetensors

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

model-00013-of-00014.safetensors

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

model-00014-of-00014.safetensors

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

preprocessor_config.json

@@ -0,0 +1,33 @@
+{
+  "crop_size": null,
+  "data_format": "channels_first",
+  "default_to_square": true,
+  "device": null,
+  "do_center_crop": null,
+  "do_convert_rgb": true,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "Llama4ImageProcessorFast",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "input_data_format": null,
+  "max_patches": 16,
+  "processor_class": "Llama4Processor",
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "resize_to_max_canvas": false,
+  "return_tensors": null,
+  "size": {
+    "height": 336,
+    "width": 336
+  }
+}

processor_config.json

@@ -0,0 +1,6 @@
+{
+  "fake_image_token": "<|image|>",
+  "image_token": "<|image|>",
+  "patch_size": 14,
+  "processor_class": "Llama4Processor"
+}

recipe.yaml

@@ -0,0 +1,11 @@
+default_stage:
+  default_modifiers:
+    GPTQModifier:
+      dampening_frac: 0.05
+      config_groups:
+        config_group:
+          targets: [Linear]
+          weights: {num_bits: 4, type: int, symmetric: true, group_size: 128, strategy: group,
+            dynamic: false, actorder: weight, observer: minmax}
+      targets: Linear
+      ignore: ['re:.*lm_head', 're:.*self_attn', 're:.*router', 're:.*vision_model', 're:.*multi_modal_projector']

special_tokens_map.json

@@ -0,0 +1,23 @@
+{
+  "bos_token": {
+    "content": "<|begin_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|eot|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|finetune_right_pad_id|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.json

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