Jboadu's picture
End of training
6453b29 verified
|
Raw
History Blame Contribute Delete
7.07 kB
metadata
library_name: transformers
license: apache-2.0
base_model: Jboadu/test-model-1-pretrain
tags:
  - axolotl
  - generated_from_trainer
datasets:
  - tool_calling/fundsTransferTool.jsonl
  - tool_calling/cardlessWithdral.jsonl
  - tool_calling/bankToWallet.jsonl
  - tool_calling/proxyPay.jsonl
  - tool_calling/forexRates.jsonl
  - cfactual/combined_all_0.jsonl
  - cfactual/combined_all_2.jsonl
  - cfactual/combined_all_6.jsonl
  - cfactual/combined_all_4.jsonl
  - cfactual/combined_all_3.jsonl
  - cfactual/combined_all_1.jsonl
  - cfactual/combined_all_5.jsonl
  - cfactual/combined_all_7.jsonl
  - cfactual/allToolCallRejections.jsonl
model-index:
  - name: test-model-1-toolcall-sharegpt
    results: []

Built with Axolotl

See axolotl config

axolotl version: 0.8.0.dev0

base_model: Jboadu/test-model-1-pretrain
tokenizer_type: AutoTokenizer
model_type: AutoModelForCausalLM
load_in_8bit: false
load_in_4bit: false
strict: false
datasets:
  - path: tool_calling/fundsTransferTool.jsonl
    type: chat_template
    field_messages: messages
    field_tools: tools
    message_property_mappings:
      role: role
      content: content
      tool_calls: tool_calls
  - path: tool_calling/cardlessWithdral.jsonl
    type: chat_template
    field_messages: messages
    field_tools: tools
    message_property_mappings:
      role: role
      content: content
      tool_calls: tool_calls
  - path: tool_calling/bankToWallet.jsonl
    type: chat_template
    field_messages: messages
    field_tools: tools
    message_property_mappings:
      role: role
      content: content
      tool_calls: tool_calls
  - path: tool_calling/proxyPay.jsonl
    type: chat_template
    field_messages: messages
    field_tools: tools
    message_property_mappings:
      role: role
      content: content
      tool_calls: tool_calls
  - path: tool_calling/forexRates.jsonl
    type: chat_template
    field_messages: messages
    field_tools: tools
    message_property_mappings:
      role: role
      content: content
      tool_calls: tool_calls
  - path: cfactual/combined_all_0.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/combined_all_2.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/combined_all_6.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/combined_all_4.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/combined_all_3.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/combined_all_1.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/combined_all_5.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/combined_all_7.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
  - path: cfactual/allToolCallRejections.jsonl
    type: chat_template
    field_messages: conversations
    message_property_mappings:
      role: from
      content: value
dataset_prepared_path: last_finetune_prepared
output_dir: ./finetune-model-output
seed: 1337
sequence_len: 5000
sample_packing: true
pad_to_sequence_len: false
shuffle_merged_datasets: true
gradient_accumulation_steps: 50
micro_batch_size: 2
eval_batch_size: 2
num_epochs: 5
optimizer: paged_adamw_8bit
lr_scheduler: constant
learning_rate: 2.0e-05
noisy_embedding_alpha: 5
weight_decay: 0
train_on_inputs: false
group_by_length: false
bf16: true
fp16: false
tf32: false
gradient_checkpointing: true
logging_steps: 1
xformers_attention: false
flash_attention: false
chat_template: chatml
auto_resume_from_checkpoints: false
warmup_ratio: 0.1
evals_per_epoch: 1
val_set_size: 0.04
saves_per_epoch: 1
eval_sample_packing: false
save_total_limit: 2
special_tokens:
  pad_token: <unk>
use_liger_kernel: true
plugins:
  - axolotl.integrations.liger.LigerPlugin
liger_rope: true
liger_rms_norm: true
liger_glu_activation: true
liger_layer_norm: true
liger_fused_linear_cross_entropy: true
sequence_length: 10000
wandb_project: test-project
wandb_entity: ""
wandb_watch: ""
wandb_run_id: ""
wandb_log_model: ""
hub_model_id: Jboadu/test-model-1-toolcall-sharegpt
hub_strategy: all_checkpoints

test-model-1-toolcall-sharegpt

This model is a fine-tuned version of Jboadu/test-model-1-pretrain on the tool_calling/fundsTransferTool.jsonl, the tool_calling/cardlessWithdral.jsonl, the tool_calling/bankToWallet.jsonl, the tool_calling/proxyPay.jsonl, the tool_calling/forexRates.jsonl, the cfactual/combined_all_0.jsonl, the cfactual/combined_all_2.jsonl, the cfactual/combined_all_6.jsonl, the cfactual/combined_all_4.jsonl, the cfactual/combined_all_3.jsonl, the cfactual/combined_all_1.jsonl, the cfactual/combined_all_5.jsonl, the cfactual/combined_all_7.jsonl and the cfactual/allToolCallRejections.jsonl datasets. It achieves the following results on the evaluation set:

  • Loss: 0.6345

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 2
  • eval_batch_size: 2
  • seed: 1337
  • gradient_accumulation_steps: 50
  • total_train_batch_size: 100
  • optimizer: Use OptimizerNames.PAGED_ADAMW_8BIT with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: constant
  • lr_scheduler_warmup_steps: 2
  • num_epochs: 5.0

Training results

Training Loss Epoch Step Validation Loss
1.3253 0.3311 1 2.8240
1.9234 0.9934 3 1.4336
1.0412 1.6623 6 0.8672
0.8892 2.3311 9 0.7712
0.0133 3.0 12 0.7462
0.5082 3.9934 15 0.6345

Framework versions

  • Transformers 4.49.0
  • Pytorch 2.5.1+cu124
  • Datasets 3.2.0
  • Tokenizers 0.21.0