Pushing merged model to Hugging Face Hub

README.md

@@ -1,199 +1,48 @@
 ---
+base_model:
+- google/gemma-3-1b-pt
+- TheMockingJay1013/gemma-3-sft-peft-dare
+- TheMockingJay1013/gemma-3-peft-safe
 library_name: transformers
-tags: []
----
-
-# Model Card for Model ID
-
-<!-- Provide a quick summary of what the model is/does. -->
-
-
-
-## Model Details
-
-### Model Description
-
-<!-- Provide a longer summary of what this model is. -->
-
-This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
-
-- **Developed by:** [More Information Needed]
-- **Funded by [optional]:** [More Information Needed]
-- **Shared by [optional]:** [More Information Needed]
-- **Model type:** [More Information Needed]
-- **Language(s) (NLP):** [More Information Needed]
-- **License:** [More Information Needed]
-- **Finetuned from model [optional]:** [More Information Needed]
-
-### Model Sources [optional]
-
-<!-- Provide the basic links for the model. -->
-
-- **Repository:** [More Information Needed]
-- **Paper [optional]:** [More Information Needed]
-- **Demo [optional]:** [More Information Needed]
-
-## Uses
-
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
-
-### Direct Use
-
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
-
-[More Information Needed]
-
-### Downstream Use [optional]
-
-<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
-
-[More Information Needed]
-
-### Out-of-Scope Use
-
-<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
-
-[More Information Needed]
-
-## Bias, Risks, and Limitations
-
-<!-- This section is meant to convey both technical and sociotechnical limitations. -->
-
-[More Information Needed]
-
-### Recommendations
-
-<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
-
-Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
-
-## How to Get Started with the Model
-
-Use the code below to get started with the model.
-
-[More Information Needed]
-
-## Training Details
-
-### Training Data
-
-<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-
-[More Information Needed]
-
-### Training Procedure
-
-<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
-
-#### Preprocessing [optional]
-
-[More Information Needed]
-
-
-#### Training Hyperparameters
-
-- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
-
-#### Speeds, Sizes, Times [optional]
-
-<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
-
-[More Information Needed]
-
-## Evaluation
+tags:
+- mergekit
+- merge
 
-<!-- This section describes the evaluation protocols and provides the results. -->
-
-### Testing Data, Factors & Metrics
-
-#### Testing Data
-
-<!-- This should link to a Dataset Card if possible. -->
-
-[More Information Needed]
-
-#### Factors
-
-<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
-
-[More Information Needed]
-
-#### Metrics
-
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
-
-[More Information Needed]
-
-### Results
-
-[More Information Needed]
-
-#### Summary
-
-
-
-## Model Examination [optional]
-
-<!-- Relevant interpretability work for the model goes here -->
-
-[More Information Needed]
-
-## Environmental Impact
-
-<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
-
-Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
-
-- **Hardware Type:** [More Information Needed]
-- **Hours used:** [More Information Needed]
-- **Cloud Provider:** [More Information Needed]
-- **Compute Region:** [More Information Needed]
-- **Carbon Emitted:** [More Information Needed]
-
-## Technical Specifications [optional]
-
-### Model Architecture and Objective
-
-[More Information Needed]
-
-### Compute Infrastructure
-
-[More Information Needed]
-
-#### Hardware
-
-[More Information Needed]
-
-#### Software
-
-[More Information Needed]
-
-## Citation [optional]
-
-<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
-
-**BibTeX:**
-
-[More Information Needed]
-
-**APA:**
-
-[More Information Needed]
-
-## Glossary [optional]
-
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
-
-[More Information Needed]
-
-## More Information [optional]
-
-[More Information Needed]
-
-## Model Card Authors [optional]
-
-[More Information Needed]
-
-## Model Card Contact
-
-[More Information Needed]
+---
+# merged
+
+This is a merge of pre-trained language models created using [mergekit](https://github.com/cg123/mergekit).
+
+## Merge Details
+### Merge Method
+
+This model was merged using the [Task Arithmetic](https://arxiv.org/abs/2212.04089) merge method using [google/gemma-3-1b-pt](https://huggingface.co/google/gemma-3-1b-pt) as a base.
+
+### Models Merged
+
+The following models were included in the merge:
+* [TheMockingJay1013/gemma-3-sft-peft-dare](https://huggingface.co/TheMockingJay1013/gemma-3-sft-peft-dare)
+* [TheMockingJay1013/gemma-3-peft-safe](https://huggingface.co/TheMockingJay1013/gemma-3-peft-safe)
+
+### Configuration
+
+The following YAML configuration was used to produce this model:
+
+```yaml
+base_model: google/gemma-3-1b-pt
+merge_method: task_arithmetic
+modules:
+  default:
+    slices:
+    - sources:
+      - layer_range: [0, 26]
+        model: TheMockingJay1013/gemma-3-sft-peft-dare
+        parameters:
+          weight: 1.0
+      - layer_range: [0, 26]
+        model: TheMockingJay1013/gemma-3-peft-safe
+        parameters:
+          weight: 1.0
+      - layer_range: [0, 26]
+        model: google/gemma-3-1b-pt
+```

added_tokens.json

@@ -1,5 +1,3 @@
 {
-  "<image_soft_token>": 262144,
-  "<|im_end|>": 262146,
-  "<|im_start|>": 262145
+  "<image_soft_token>": 262144
 }

config.json

@@ -5,9 +5,9 @@
   "attention_bias": false,
   "attention_dropout": 0.0,
   "attn_logit_softcapping": null,
-  "bos_token_id": 262145,
+  "bos_token_id": 2,
   "cache_implementation": "hybrid",
-  "eos_token_id": 262146,
+  "eos_token_id": 1,
   "final_logit_softcapping": null,
   "head_dim": 256,
   "hidden_activation": "gelu_pytorch_tanh",
@@ -19,7 +19,7 @@
   "num_attention_heads": 4,
   "num_hidden_layers": 26,
   "num_key_value_heads": 1,
-  "pad_token_id": 262146,
+  "pad_token_id": 0,
   "query_pre_attn_scalar": 256,
   "rms_norm_eps": 1e-06,
   "rope_local_base_freq": 10000,
@@ -27,8 +27,8 @@
   "rope_theta": 1000000,
   "sliding_window": 512,
   "sliding_window_pattern": 6,
-  "torch_dtype": "float32",
-  "transformers_version": "4.50.3",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.1",
   "use_cache": true,
-  "vocab_size": 262147
+  "vocab_size": 262144
 }

mergekit_config.yml

@@ -0,0 +1,16 @@
+base_model: google/gemma-3-1b-pt
+merge_method: task_arithmetic
+modules:
+  default:
+    slices:
+    - sources:
+      - layer_range: [0, 26]
+        model: TheMockingJay1013/gemma-3-sft-peft-dare
+        parameters:
+          weight: 1.0
+      - layer_range: [0, 26]
+        model: TheMockingJay1013/gemma-3-peft-safe
+        parameters:
+          weight: 1.0
+      - layer_range: [0, 26]
+        model: google/gemma-3-1b-pt

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dcca37f1322b24c79e5a10f1bdeb64a0bd6b06368959208e8cda4c48cf38cedc
-size 3999596784
+oid sha256:26b87f21d3cb75d8d7e394fb37a14e1c58def00c1e45ee9a675a8d01adee38a5
+size 2603791264

special_tokens_map.json

@@ -1,26 +1,28 @@
 {
-  "additional_special_tokens": [
-    {
-      "content": "<|im_start|>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false
-    },
-    {
-      "content": "<|im_end|>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false
-    }
-  ],
   "boi_token": "<start_of_image>",
-  "bos_token": "<|im_start|>",
+  "bos_token": {
+    "content": "<bos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
   "eoi_token": "<end_of_image>",
-  "eos_token": "<|im_end|>",
+  "eos_token": {
+    "content": "<eos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
   "image_token": "<image_soft_token>",
-  "pad_token": "<|im_end|>",
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
   "unk_token": {
     "content": "<unk>",
     "lstrip": false,

tokenizer.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3c5591c1449f71a5ae4c2ed1cb1cbdadf42e4cc9537c9850184d520a42cb18dc
-size 33384944
+oid sha256:7d4046bf0505a327dd5a0abbb427ecd4fc82f99c2ceaa170bc61ecde12809b0c
+size 33384570

tokenizer_config.json

@@ -2160,7 +2160,7 @@
       "normalized": false,
       "rstrip": false,
       "single_word": false,
-      "special": true
+      "special": false
     },
     "256000": {
       "content": "<end_of_image>",
@@ -2168,7 +2168,7 @@
       "normalized": false,
       "rstrip": false,
       "single_word": false,
-      "special": true
+      "special": false
     },
     "256001": {
       "content": "<unused99>",
@@ -51321,34 +51321,13 @@
       "rstrip": false,
       "single_word": false,
       "special": true
-    },
-    "262145": {
-      "content": "<|im_start|>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
-    },
-    "262146": {
-      "content": "<|im_end|>",
-      "lstrip": false,
-      "normalized": false,
-      "rstrip": false,
-      "single_word": false,
-      "special": true
     }
   },
-  "additional_special_tokens": [
-    "<|im_start|>",
-    "<|im_end|>"
-  ],
   "boi_token": "<start_of_image>",
-  "bos_token": "<|im_start|>",
-  "chat_template": "{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
+  "bos_token": "<bos>",
   "clean_up_tokenization_spaces": false,
   "eoi_token": "<end_of_image>",
-  "eos_token": "<|im_end|>",
+  "eos_token": "<eos>",
   "extra_special_tokens": {
     "boi_token": "<start_of_image>",
     "eoi_token": "<end_of_image>",
@@ -51356,7 +51335,7 @@
   },
   "image_token": "<image_soft_token>",
   "model_max_length": 1000000000000000019884624838656,
-  "pad_token": "<|im_end|>",
+  "pad_token": "<pad>",
   "sp_model_kwargs": null,
   "spaces_between_special_tokens": false,
   "tokenizer_class": "GemmaTokenizer",