Upload model files

.gitattributes

@@ -33,3 +33,5 @@ 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,675 @@
+{
+  "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": {
+          "actorder": null,
+          "block_structure": null,
+          "dynamic": "local",
+          "group_size": 16,
+          "num_bits": 4,
+          "observer": "minmax",
+          "observer_kwargs": {},
+          "strategy": "tensor_group",
+          "symmetric": true,
+          "type": "float"
+        },
+        "output_activations": null,
+        "targets": [
+          "Linear"
+        ],
+        "weights": {
+          "actorder": null,
+          "block_structure": null,
+          "dynamic": false,
+          "group_size": 16,
+          "num_bits": 4,
+          "observer": "minmax",
+          "observer_kwargs": {},
+          "strategy": "tensor_group",
+          "symmetric": true,
+          "type": "float"
+        }
+      }
+    },
+    "format": "nvfp4-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,
+    "cache_implementation": "hybrid",
+    "eos_token_id": [
+      200001,
+      200007,
+      200008
+    ],
+    "floor_scale": 8192,
+    "for_llm_compressor": false,
+    "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": 262144,
+    "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.3",
+  "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,
+    "torch_dtype": "bfloat16",
+    "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.3"
+}

model-00001-of-00014.safetensors

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

model-00002-of-00014.safetensors

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

model-00003-of-00014.safetensors

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

model-00004-of-00014.safetensors

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

model-00005-of-00014.safetensors

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

model-00006-of-00014.safetensors

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

model-00007-of-00014.safetensors

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

model-00008-of-00014.safetensors

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

model-00009-of-00014.safetensors

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

model-00010-of-00014.safetensors

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

model-00011-of-00014.safetensors

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

model-00012-of-00014.safetensors

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

model-00013-of-00014.safetensors

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

model-00014-of-00014.safetensors

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

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,7 @@
+default_stage:
+  default_modifiers:
+    QuantizationModifier:
+      targets: [Linear]
+      ignore: ['re:.*lm_head', 're:.*self_attn', 're:.*router', 're:.*vision_model', 're:.*multi_modal_projector',
+        're:.*multi_modal_projector', Llama4TextAttention]
+      scheme: NVFP4

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|>",
+    "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