Upload InkubaLM multi-task with adapters distill + quantize

config.json

@@ -0,0 +1,158 @@
+{
+  "_name_or_path": "YvanCarre/InkubaLM-multitask_dist",
+  "adapters": {
+    "adapters": {
+      "machine-translation": "26cd1b10db746518",
+      "sentiment-analysis": "seq_bn",
+      "xnli": "seq_bn"
+    },
+    "config_map": {
+      "26cd1b10db746518": {
+        "adapter_residual_before_ln": false,
+        "cross_adapter": false,
+        "dropout": 0.0,
+        "factorized_phm_W": true,
+        "factorized_phm_rule": false,
+        "hypercomplex_nonlinearity": "glorot-uniform",
+        "init_weights": "bert",
+        "inv_adapter": null,
+        "inv_adapter_reduction_factor": null,
+        "is_parallel": false,
+        "learn_phm": true,
+        "leave_out": [],
+        "ln_after": false,
+        "ln_before": false,
+        "mh_adapter": false,
+        "non_linearity": "relu",
+        "original_ln_after": true,
+        "original_ln_before": true,
+        "output_adapter": true,
+        "phm_bias": true,
+        "phm_c_init": "normal",
+        "phm_dim": 4,
+        "phm_init_range": 0.0001,
+        "phm_layer": false,
+        "phm_rank": 1,
+        "reduction_factor": 8,
+        "residual_before_ln": true,
+        "scaling": 1.0,
+        "shared_W_phm": false,
+        "shared_phm_rule": true,
+        "stochastic_depth": 0.0,
+        "use_gating": false
+      }
+    },
+    "fusion_config_map": {},
+    "fusion_name_map": {},
+    "fusions": {}
+  },
+  "architectures": [
+    "LlamaAdapterModel"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoModelForCausalLM": "lelapa/InkubaLM-0.4B--vulavulaslm.VulavulaLlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "head_dim": 64,
+  "hidden_act": "silu",
+  "hidden_size": 1024,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 2752,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2
+  },
+  "max_position_embeddings": 2048,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 4,
+  "num_key_value_heads": 32,
+  "prediction_heads": {
+    "default": {
+      "activation_function": null,
+      "bias": false,
+      "dropout_prob": null,
+      "embedding_size": 2048,
+      "head_type": "causal_lm",
+      "label2id": null,
+      "layer_norm": false,
+      "layers": 1,
+      "shift_labels": true,
+      "vocab_size": 61788
+    },
+    "machine-translation": {
+      "activation_function": "gelu",
+      "bias": true,
+      "dropout_prob": null,
+      "embedding_size": 2048,
+      "head_type": "causal_lm",
+      "label2id": null,
+      "layer_norm": true,
+      "layers": 2,
+      "shift_labels": true,
+      "vocab_size": 61788
+    },
+    "sentiment-analysis": {
+      "activation_function": "tanh",
+      "bias": true,
+      "dropout_prob": null,
+      "head_type": "classification",
+      "label2id": {
+        "LABEL_0": 0,
+        "LABEL_1": 1,
+        "LABEL_2": 2
+      },
+      "layers": 2,
+      "num_labels": 3,
+      "use_pooler": false
+    },
+    "xnli": {
+      "activation_function": "tanh",
+      "bias": true,
+      "dropout_prob": null,
+      "head_type": "classification",
+      "label2id": {
+        "LABEL_0": 0,
+        "LABEL_1": 1,
+        "LABEL_2": 2
+      },
+      "layers": 2,
+      "num_labels": 3,
+      "use_pooler": false
+    }
+  },
+  "pretraining_tp": 1,
+  "quantization_config": {
+    "_load_in_4bit": true,
+    "_load_in_8bit": false,
+    "bnb_4bit_compute_dtype": "float16",
+    "bnb_4bit_quant_storage": "uint8",
+    "bnb_4bit_quant_type": "nf4",
+    "bnb_4bit_use_double_quant": true,
+    "llm_int8_enable_fp32_cpu_offload": false,
+    "llm_int8_has_fp16_weight": false,
+    "llm_int8_skip_modules": null,
+    "llm_int8_threshold": 6.0,
+    "load_in_4bit": true,
+    "load_in_8bit": false,
+    "quant_method": "bitsandbytes"
+  },
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.47.1",
+  "use_cache": true,
+  "vocab_size": 61788
+}

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "transformers_version": "4.47.1"
+}

machine-translation/adapter_config.json

@@ -0,0 +1,42 @@
+{
+  "config": {
+    "adapter_residual_before_ln": false,
+    "cross_adapter": false,
+    "dropout": 0.0,
+    "factorized_phm_W": true,
+    "factorized_phm_rule": false,
+    "hypercomplex_nonlinearity": "glorot-uniform",
+    "init_weights": "bert",
+    "inv_adapter": null,
+    "inv_adapter_reduction_factor": null,
+    "is_parallel": false,
+    "learn_phm": true,
+    "leave_out": [],
+    "ln_after": false,
+    "ln_before": false,
+    "mh_adapter": false,
+    "non_linearity": "relu",
+    "original_ln_after": true,
+    "original_ln_before": true,
+    "output_adapter": true,
+    "phm_bias": true,
+    "phm_c_init": "normal",
+    "phm_dim": 4,
+    "phm_init_range": 0.0001,
+    "phm_layer": false,
+    "phm_rank": 1,
+    "reduction_factor": 8,
+    "residual_before_ln": true,
+    "scaling": 1.0,
+    "shared_W_phm": false,
+    "shared_phm_rule": true,
+    "stochastic_depth": 0.0,
+    "use_gating": false
+  },
+  "hidden_size": 1024,
+  "model_class": "LlamaAdapterModel",
+  "model_name": "YvanCarre/InkubaLM-multitask_dist",
+  "model_type": "llama",
+  "name": "machine-translation",
+  "version": "adapters.1.1.0"
+}

machine-translation/head_config.json

@@ -0,0 +1,20 @@
+{
+  "config": {
+    "activation_function": "gelu",
+    "bias": true,
+    "dropout_prob": null,
+    "embedding_size": 2048,
+    "head_type": "causal_lm",
+    "label2id": null,
+    "layer_norm": true,
+    "layers": 2,
+    "shift_labels": true,
+    "vocab_size": 61788
+  },
+  "hidden_size": 1024,
+  "model_class": "LlamaAdapterModel",
+  "model_name": "YvanCarre/InkubaLM-multitask_dist",
+  "model_type": "llama",
+  "name": "machine-translation",
+  "version": "adapters.1.1.0"
+}

machine-translation/pytorch_adapter.bin

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

machine-translation/pytorch_model_head.bin

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

model.safetensors

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

sentiment-analysis/adapter_config.json

@@ -0,0 +1,42 @@
+{
+  "config": {
+    "adapter_residual_before_ln": false,
+    "cross_adapter": false,
+    "dropout": 0.0,
+    "factorized_phm_W": true,
+    "factorized_phm_rule": false,
+    "hypercomplex_nonlinearity": "glorot-uniform",
+    "init_weights": "bert",
+    "inv_adapter": null,
+    "inv_adapter_reduction_factor": null,
+    "is_parallel": false,
+    "learn_phm": true,
+    "leave_out": [],
+    "ln_after": false,
+    "ln_before": false,
+    "mh_adapter": false,
+    "non_linearity": "relu",
+    "original_ln_after": true,
+    "original_ln_before": true,
+    "output_adapter": true,
+    "phm_bias": true,
+    "phm_c_init": "normal",
+    "phm_dim": 4,
+    "phm_init_range": 0.0001,
+    "phm_layer": false,
+    "phm_rank": 1,
+    "reduction_factor": 16,
+    "residual_before_ln": true,
+    "scaling": 1.0,
+    "shared_W_phm": false,
+    "shared_phm_rule": true,
+    "stochastic_depth": 0.0,
+    "use_gating": false
+  },
+  "hidden_size": 1024,
+  "model_class": "LlamaAdapterModel",
+  "model_name": "YvanCarre/InkubaLM-multitask_dist",
+  "model_type": "llama",
+  "name": "sentiment-analysis",
+  "version": "adapters.1.1.0"
+}

sentiment-analysis/head_config.json

@@ -0,0 +1,22 @@
+{
+  "config": {
+    "activation_function": "tanh",
+    "bias": true,
+    "dropout_prob": null,
+    "head_type": "classification",
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1,
+      "LABEL_2": 2
+    },
+    "layers": 2,
+    "num_labels": 3,
+    "use_pooler": false
+  },
+  "hidden_size": 1024,
+  "model_class": "LlamaAdapterModel",
+  "model_name": "YvanCarre/InkubaLM-multitask_dist",
+  "model_type": "llama",
+  "name": "sentiment-analysis",
+  "version": "adapters.1.1.0"
+}

sentiment-analysis/pytorch_adapter.bin

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

sentiment-analysis/pytorch_model_head.bin

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

special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "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": "</s>",
+  "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:1c41fcc6d44fcc4e8269e41dffe0123687baf800bd95a9c8b5d48abd9cb8971b
+size 991189

tokenizer_config.json

@@ -0,0 +1,43 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "add_prefix_space": true,
+  "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
+    }
+  },
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "extra_special_tokens": {},
+  "legacy": true,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "</s>",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

xnli/adapter_config.json

@@ -0,0 +1,42 @@
+{
+  "config": {
+    "adapter_residual_before_ln": false,
+    "cross_adapter": false,
+    "dropout": 0.0,
+    "factorized_phm_W": true,
+    "factorized_phm_rule": false,
+    "hypercomplex_nonlinearity": "glorot-uniform",
+    "init_weights": "bert",
+    "inv_adapter": null,
+    "inv_adapter_reduction_factor": null,
+    "is_parallel": false,
+    "learn_phm": true,
+    "leave_out": [],
+    "ln_after": false,
+    "ln_before": false,
+    "mh_adapter": false,
+    "non_linearity": "relu",
+    "original_ln_after": true,
+    "original_ln_before": true,
+    "output_adapter": true,
+    "phm_bias": true,
+    "phm_c_init": "normal",
+    "phm_dim": 4,
+    "phm_init_range": 0.0001,
+    "phm_layer": false,
+    "phm_rank": 1,
+    "reduction_factor": 16,
+    "residual_before_ln": true,
+    "scaling": 1.0,
+    "shared_W_phm": false,
+    "shared_phm_rule": true,
+    "stochastic_depth": 0.0,
+    "use_gating": false
+  },
+  "hidden_size": 1024,
+  "model_class": "LlamaAdapterModel",
+  "model_name": "YvanCarre/InkubaLM-multitask_dist",
+  "model_type": "llama",
+  "name": "xnli",
+  "version": "adapters.1.1.0"
+}

xnli/head_config.json

@@ -0,0 +1,22 @@
+{
+  "config": {
+    "activation_function": "tanh",
+    "bias": true,
+    "dropout_prob": null,
+    "head_type": "classification",
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1,
+      "LABEL_2": 2
+    },
+    "layers": 2,
+    "num_labels": 3,
+    "use_pooler": false
+  },
+  "hidden_size": 1024,
+  "model_class": "LlamaAdapterModel",
+  "model_name": "YvanCarre/InkubaLM-multitask_dist",
+  "model_type": "llama",
+  "name": "xnli",
+  "version": "adapters.1.1.0"
+}

xnli/pytorch_adapter.bin

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

xnli/pytorch_model_head.bin

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