|
--- |
|
library_name: transformers |
|
license: llama3.2 |
|
base_model: macadeliccc/magistrate-3.2-3b-base |
|
datasets: |
|
- teknium/OpenHermes-2.5 |
|
- NousResearch/hermes-function-calling-v1 |
|
- arcee-ai/The-Tome |
|
- cognitivecomputations/SystemChat-2.0 |
|
tags: |
|
- spectrum |
|
- llama-3 |
|
- axolotl |
|
- legal |
|
- HFforLegal |
|
language: |
|
- en |
|
pipeline_tag: text-generation |
|
--- |
|
# magistrate-3.2-3b-it |
|
|
|
This model is a fine-tuned version of [macadeliccc/magistrate-3.2-3b-base](https://huggingface.co/macadeliccc/magistrate-3.2-3b-base) on the None dataset. |
|
It achieves the following results on the evaluation set: |
|
- Loss: 0.8067 |
|
|
|
<!-- This model card has been generated automatically according to the information the Trainer had access to. You |
|
should probably proofread and complete it, then remove this comment. --> |
|
|
|
[<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl) |
|
<details><summary>See axolotl config</summary> |
|
|
|
axolotl version: `0.4.1` |
|
```yaml |
|
base_model: macadeliccc/magistrate-3.2-3b-base |
|
model_type: LlamaForCausalLM |
|
tokenizer_type: AutoTokenizer |
|
|
|
load_in_8bit: false |
|
load_in_4bit: false |
|
strict: false |
|
|
|
datasets: |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/hermes-2.5.jsonl |
|
# - path: json |
|
# type: sharegpt |
|
# conversation: chatml |
|
# data_files: train/financial_instructions_cleaned_2.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/glaive-function-calling-5k.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/func-calling-singleturn.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/func-calling.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/json-mode-agentic.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/json-mode-singleturn.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/reasoning_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/systemchat_2_0_small.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/303_creative_llc_v__elenis_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/abitron_austria_gmbh_v__hetronic_international__inc__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/acheson_hotels__llc_v__laufer_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/alexander_v__sc_conference_of_naacp_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/amgen_inc__v__sanofi_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/andy_warhol_found___inc__v__goldsmith_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/arizona_v__navajo_nation_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/becerra__sec__of_h_hs_v__san_carlos_apache_tribe_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/biden_v__nebraska_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/bissonnette_v__lepage_bakeries_park_st___llc_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/bittner_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/brown_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/cantero_v__bank_of_america__n_a__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/cfpb_v__com__fin__services_assn__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/chiaverini_v__city_of_napoleon_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/ciminelli_v__united_state_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/city_of_grants_pass_v__johnson_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/coinbase__inc__v__bielski_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/coinbase__inc__v__suski_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/connelly_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/corner_post__inc__v__bd__of_governors__frs_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/counterman_v__colorado_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/cruz_v__arizona_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/culley_v__marshall_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/dept__of_agric__rural_dev__v__kirtz_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/dept__of_education_v__brown_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/dept__of_state_v__munoz_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/devillier_v__texas_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/diaz_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/dubin_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/dupree_v__younger_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/erlinger_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/fbi_v__fikre_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/fda_v__alliance_hippocratic_medicine_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/financial_oversight_board_v__cpi_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/fischer_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/garland__att_y_gen__v__cargill_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/glacier_northwest__inc__v__int_l_brotherhood_of_teamsters_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/gonzalez_v__google_llc_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/gonzalez_v__trevino_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/great_lakes_insurance_se_v__raiders_retreat_realty_co___llc_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/groff_v__dejoy_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/harrington_v__purdue_pharma_l_p__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/harrow_v__dept__of_defense_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/health_and_hospital_corp__v__talevski_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/helix_energy_solutions_v__hewitt_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/in_re_grand_jury_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/jack_daniel_s_properties__inc__v__vip_products_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/jones_v__hendrix_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/karcho_polselli_v__irs_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/lac_du_flambeau_band_v__coughlin_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/lindke_v__freed_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/loper_bright_enterprises__inc__v__raimondo__sec__of_comm__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/lora_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/macquarie_infrastructure_corp__v__moab_partners__l_p__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/mallory_v__norfolk_southern_railway_co__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/mcintosh_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/merrill_v__milligan_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/moore_v__harper_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/moore_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/moyle_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/muldrow_v__st__louis_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/murray_v__ubs_securities__llc_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/murthy__surgeon_gen__v__missouri_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/netchoice__llc_v__paxton_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/new_york_v__new_jersey_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/nra_v__vullo_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/o_connor_ratcliff_v__garnier_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/oh_adjutant_gen__s_dept__v__flra_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/ohio_v__epa_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/perez_v__sturgis_public_schools_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/pugin_v__garland_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/pulsifer_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/relentless__inc__v__dept__of_commerce_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/rudisill_v__mcdonough__sec__of_va_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/sackett_v__epa_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/samia_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/santos_zacaria_v__garland__att_y_gen__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/sec_v__cochran_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/sec_v__jarkesy_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/sheetz_v__county_of_el_dorado_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/slack_technologies__llc_v__pirani_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/smith_v__arizona_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/smith_v__spizzirri_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/smith_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/snyder_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/starbucks_corp__v__mckinney_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/students_for_fair_admissions_v__university_of_nc_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/texas_v__new_mexico_and_colorado_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/thornell_v__jones_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/truck_insurance_exchange_v__kaiser_gypsum_co__inc__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/trump_v__anderson_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/turkiye_halk_bankasi_a_s__v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/twitter__inc__v__taamneh_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/tyler_v__hennepin_county_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/u_s___ex_rel__polansky_v__executive_health_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/u_s___ex_rel__schutte_v__supervalu_inc__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/united_states_trustee_v__john_q__hammons_fall_2006__llc_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/united_states_v__hansen_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/united_states_v__rahimi_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/united_states_v__texas_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/vidal__under_sec__of_comm__v__elster_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/warner_chappell_music__inc__v__nealy_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/wilkins_v__united_states_sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/wilkinson_v__garland__att_y_gen__sharegpt.json |
|
- path: json |
|
type: sharegpt |
|
conversation: chatml |
|
data_files: train/argument_dataset/yegiazaryan_v__smagin_sharegpt.json |
|
|
|
chat_template: chatml |
|
|
|
unfrozen_parameters: |
|
- ^lm_head.weight$ |
|
- ^model.embed_tokens.weight$ |
|
# input_layernorm layers |
|
- model.layers.0.input_layernorm |
|
- model.layers.1.input_layernorm |
|
- model.layers.2.input_layernorm |
|
- model.layers.3.input_layernorm |
|
- model.layers.4.input_layernorm |
|
- model.layers.5.input_layernorm |
|
- model.layers.6.input_layernorm |
|
- model.layers.7.input_layernorm |
|
- model.layers.8.input_layernorm |
|
- model.layers.9.input_layernorm |
|
- model.layers.10.input_layernorm |
|
- model.layers.11.input_layernorm |
|
- model.layers.12.input_layernorm |
|
- model.layers.13.input_layernorm |
|
# mlp.down_proj layers |
|
- model.layers.0.mlp.down_proj |
|
- model.layers.1.mlp.down_proj |
|
- model.layers.17.mlp.down_proj |
|
- model.layers.19.mlp.down_proj |
|
- model.layers.18.mlp.down_proj |
|
- model.layers.5.mlp.down_proj |
|
- model.layers.20.mlp.down_proj |
|
- model.layers.2.mlp.down_proj |
|
- model.layers.4.mlp.down_proj |
|
- model.layers.6.mlp.down_proj |
|
- model.layers.3.mlp.down_proj |
|
- model.layers.16.mlp.down_proj |
|
- model.layers.15.mlp.down_proj |
|
- model.layers.13.mlp.down_proj |
|
# mlp.gate_proj layers |
|
- model.layers.0.mlp.gate_proj |
|
- model.layers.1.mlp.gate_proj |
|
- model.layers.2.mlp.gate_proj |
|
- model.layers.3.mlp.gate_proj |
|
- model.layers.22.mlp.gate_proj |
|
- model.layers.21.mlp.gate_proj |
|
- model.layers.20.mlp.gate_proj |
|
- model.layers.23.mlp.gate_proj |
|
- model.layers.19.mlp.gate_proj |
|
- model.layers.4.mlp.gate_proj |
|
- model.layers.18.mlp.gate_proj |
|
- model.layers.17.mlp.gate_proj |
|
- model.layers.5.mlp.gate_proj |
|
- model.layers.24.mlp.gate_proj |
|
# mlp.up_proj layers |
|
- model.layers.4.mlp.up_proj |
|
- model.layers.3.mlp.up_proj |
|
- model.layers.5.mlp.up_proj |
|
- model.layers.6.mlp.up_proj |
|
- model.layers.7.mlp.up_proj |
|
- model.layers.2.mlp.up_proj |
|
- model.layers.8.mlp.up_proj |
|
- model.layers.14.mlp.up_proj |
|
- model.layers.13.mlp.up_proj |
|
- model.layers.11.mlp.up_proj |
|
- model.layers.9.mlp.up_proj |
|
- model.layers.1.mlp.up_proj |
|
- model.layers.15.mlp.up_proj |
|
- model.layers.12.mlp.up_proj |
|
# post_attention_layernorm layers |
|
- model.layers.0.post_attention_layernorm |
|
- model.layers.1.post_attention_layernorm |
|
- model.layers.2.post_attention_layernorm |
|
- model.layers.3.post_attention_layernorm |
|
- model.layers.4.post_attention_layernorm |
|
- model.layers.5.post_attention_layernorm |
|
- model.layers.6.post_attention_layernorm |
|
- model.layers.7.post_attention_layernorm |
|
- model.layers.8.post_attention_layernorm |
|
- model.layers.9.post_attention_layernorm |
|
- model.layers.10.post_attention_layernorm |
|
- model.layers.11.post_attention_layernorm |
|
- model.layers.12.post_attention_layernorm |
|
- model.layers.13.post_attention_layernorm |
|
# self_attn.k_proj layers |
|
- model.layers.25.self_attn.k_proj |
|
- model.layers.22.self_attn.k_proj |
|
- model.layers.19.self_attn.k_proj |
|
- model.layers.20.self_attn.k_proj |
|
- model.layers.17.self_attn.k_proj |
|
- model.layers.24.self_attn.k_proj |
|
- model.layers.23.self_attn.k_proj |
|
- model.layers.18.self_attn.k_proj |
|
- model.layers.21.self_attn.k_proj |
|
- model.layers.27.self_attn.k_proj |
|
- model.layers.15.self_attn.k_proj |
|
- model.layers.10.self_attn.k_proj |
|
- model.layers.6.self_attn.k_proj |
|
- model.layers.5.self_attn.k_proj |
|
# self_attn.o_proj layers |
|
- model.layers.13.self_attn.o_proj |
|
- model.layers.7.self_attn.o_proj |
|
- model.layers.12.self_attn.o_proj |
|
- model.layers.10.self_attn.o_proj |
|
- model.layers.5.self_attn.o_proj |
|
- model.layers.21.self_attn.o_proj |
|
- model.layers.6.self_attn.o_proj |
|
- model.layers.19.self_attn.o_proj |
|
- model.layers.8.self_attn.o_proj |
|
- model.layers.20.self_attn.o_proj |
|
- model.layers.22.self_attn.o_proj |
|
- model.layers.9.self_attn.o_proj |
|
- model.layers.17.self_attn.o_proj |
|
- model.layers.11.self_attn.o_proj |
|
# self_attn.q_proj layers |
|
- model.layers.12.self_attn.q_proj |
|
- model.layers.13.self_attn.q_proj |
|
- model.layers.9.self_attn.q_proj |
|
- model.layers.8.self_attn.q_proj |
|
- model.layers.10.self_attn.q_proj |
|
- model.layers.14.self_attn.q_proj |
|
- model.layers.11.self_attn.q_proj |
|
- model.layers.15.self_attn.q_proj |
|
- model.layers.26.self_attn.q_proj |
|
- model.layers.6.self_attn.q_proj |
|
- model.layers.7.self_attn.q_proj |
|
- model.layers.16.self_attn.q_proj |
|
- model.layers.5.self_attn.q_proj |
|
- model.layers.25.self_attn.q_proj |
|
# model.norm layers |
|
# self_attn.v_proj layers |
|
- model.layers.23.self_attn.v_proj |
|
- model.layers.14.self_attn.v_proj |
|
- model.layers.15.self_attn.v_proj |
|
- model.layers.19.self_attn.v_proj |
|
- model.layers.3.self_attn.v_proj |
|
- model.layers.18.self_attn.v_proj |
|
- model.layers.25.self_attn.v_proj |
|
- model.layers.4.self_attn.v_proj |
|
- model.layers.17.self_attn.v_proj |
|
- model.layers.22.self_attn.v_proj |
|
- model.layers.20.self_attn.v_proj |
|
- model.layers.13.self_attn.v_proj |
|
- model.layers.6.self_attn.v_proj |
|
- model.layers.27.self_attn.v_proj |
|
|
|
val_set_size: 0.05 |
|
output_dir: ./outputs/magistrate-3.2-3b |
|
|
|
sequence_len: 8192 |
|
sample_packing: true |
|
eval_sample_packing: false |
|
pad_to_sequence_len: true |
|
|
|
adapter: |
|
|
|
wandb_project: |
|
wandb_entity: |
|
wandb_watch: |
|
wandb_name: |
|
wandb_log_model: |
|
|
|
gradient_accumulation_steps: 8 |
|
micro_batch_size: 1 |
|
num_epochs: 3 |
|
optimizer: paged_adamw_32bit |
|
lr_scheduler: cosine |
|
learning_rate: 2e-4 |
|
|
|
train_on_inputs: false |
|
group_by_length: false |
|
bf16: auto |
|
fp16: |
|
tf32: false |
|
|
|
gradient_checkpointing: true |
|
early_stopping_patience: |
|
resume_from_checkpoint: |
|
local_rank: |
|
logging_steps: 1 |
|
xformers_attention: |
|
flash_attention: true |
|
s2_attention: |
|
|
|
warmup_steps: 1000 |
|
evals_per_epoch: 2 |
|
eval_table_size: |
|
eval_max_new_tokens: 128 |
|
saves_per_epoch: 1 |
|
debug: |
|
deepspeed: deepspeed_configs/zero3.json |
|
weight_decay: 0.0 |
|
fsdp: |
|
fsdp_config: |
|
special_tokens: |
|
eos_token: "<|im_end|>" |
|
pad_token: "<|end_of_text|>" |
|
tokens: |
|
- "<|im_start|>" |
|
- "<|im_end|>" |
|
``` |
|
|
|
</details><br> |
|
|
|
|
|
## Model description |
|
|
|
Magistrate-3.2-3b-it is a legal assistant specializing in US Supreme Court case law and US Federal regulations. |
|
|
|
The base model is pretrained with ~250M tokens containing no synthetic legal data. The instruct model does contain synthetic data. |
|
|
|
## Intended uses & limitations |
|
|
|
This model is for research purposes and for continued development of the legal specialty. You are liable for all model outputs. |
|
|
|
## Training and evaluation data |
|
|
|
This model was trained on a variety of standard open source datasets like OpenHermes-2.5, hermes-function-calling, and some select entries from the Tome. |
|
Additionally, I have included a comprehensive, non-synthetic argument dataset. This is a work in progress but has shown promising results so far. |
|
|
|
## Training procedure |
|
|
|
Spectrum top 35% finetune for both pretrain and SFT. Thanks to the cognitive computations team for the work done with spectrum. |
|
|
|
+ Pretraining methodology based on Cohere's paper: [To Code, or Not To Code? Exploring Impact of Code in Pre-training](https://arxiv.org/abs/2408.10914) |
|
+ Instruct finetune largely based on OpenHermes-2.5 and hermes-function-calling |
|
|
|
### Training hyperparameters |
|
|
|
The following hyperparameters were used during training: |
|
- learning_rate: 0.0002 |
|
- train_batch_size: 1 |
|
- eval_batch_size: 1 |
|
- seed: 42 |
|
- distributed_type: multi-GPU |
|
- num_devices: 2 |
|
- gradient_accumulation_steps: 8 |
|
- total_train_batch_size: 16 |
|
- total_eval_batch_size: 2 |
|
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 |
|
- lr_scheduler_type: cosine |
|
- lr_scheduler_warmup_steps: 1000 |
|
- num_epochs: 3 |
|
|
|
### Training results |
|
|
|
| Training Loss | Epoch | Step | Validation Loss | |
|
|:-------------:|:------:|:----:|:---------------:| |
|
| 1.3754 | 0.0005 | 1 | 1.7429 | |
|
| 1.0 | 0.5002 | 1017 | 0.8864 | |
|
| 0.9482 | 1.0005 | 2034 | 0.8395 | |
|
| 0.6817 | 1.4987 | 3051 | 0.8063 | |
|
| 0.697 | 1.9991 | 4068 | 0.7580 | |
|
| 0.3769 | 2.4966 | 5085 | 0.8140 | |
|
| 0.4278 | 2.9965 | 6102 | 0.8067 | |
|
|
|
|
|
### Framework versions |
|
|
|
- Transformers 4.45.0 |
|
- Pytorch 2.3.1+cu121 |
|
- Datasets 2.21.0 |
|
- Tokenizers 0.20.0 |
