mergekit_config.yml

{
  "models": [
    {
      "model": "saransh03sharma/sft_model_state_dict-peft",
      "parameters": {
        "weight": 1.0
      }
    },
    {
      "model": "saransh03sharma/sft-unsafe-model-full",
      "parameters": {
        "weight": -1.0
      }
    }
  ],
  "merge_method": "task_arithmetic",
  "base_model": "google/gemma-3-1b-pt",
  "dtype": "float16",
  "output_dir": "sft_model_peft"
}