Built with Axolotl

See axolotl config

axolotl version: 0.8.0.dev0

base_model: Qwen/Qwen3-4B
model_type: AutoModelForCausalLM
tokenizer_type: AutoTokenizer

# QLoRA
adapter: lora
lora_r: 32
lora_alpha: 64
lora_dropout: 0.05
lora_target_modules: [q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj]

datasets:
- path: laurent-maille/pcl-test-S17
  type: chat_template
  field_messages: messages
  conversation: chat
dataset_prepared_path: ./prepared/plc_sharegpt
output_dir: ./outputs/valuoty-indus-plc-4B


# Séquences
sequence_len: 4096
sample_packing: true
cutoff_len: 2048           # <- 4096 si nécessaire, sinon 2048 conseillé
packing: true
pad_to_sequence_len: false
group_by_length: true
flash_attn_impl: fa2       # mets à "none" si FA2 non dispo

# Quantization / dtypes
load_in_4bit: true         # <- remplace load_in_8bit
load_in_8bit: false
bnb_4bit_quant_type: nf4
bnb_4bit_use_double_quant: true
bnb_4bit_compute_dtype: bfloat16
bf16: true
fp16: false

# Optim & training
optimizer: adamw_bnb_8bit
micro_batch_size: 1
per_device_train_batch_size: 8     # 8–16 à 2048 tokens sur A40 48Go
per_device_eval_batch_size: 8
gradient_accumulation_steps: 16    # effectif ≈ 128 (8×16); monte à 32 pour 256
gradient_checkpointing: true
max_grad_norm: 0.3
learning_rate: 1.0e-4
weight_decay: 0.0
lr_scheduler_type: cosine
warmup_ratio: 0.03
num_train_epochs: 2

# Eval / logs / save
evaluation_strategy: steps
eval_steps: 1000
logging_steps: 20
save_strategy: steps
save_steps: 1000
save_total_limit: 4
save_safetensors: true

outputs/valuoty-indus-plc-4B

This model is a fine-tuned version of Qwen/Qwen3-4B on the laurent-maille/pcl-test-S17 dataset.

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: 0.0001
  • train_batch_size: 1
  • eval_batch_size: 1
  • seed: 42
  • gradient_accumulation_steps: 16
  • total_train_batch_size: 16
  • optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: cosine
  • lr_scheduler_warmup_steps: 2
  • num_epochs: 1.0

Training results

Framework versions

  • PEFT 0.14.0
  • Transformers 4.55.4
  • Pytorch 2.5.1+cu124
  • Datasets 3.2.0
  • Tokenizers 0.21.0
Downloads last month
1
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for laurent-maille/Valuoty-industry-plc-4B-V0.2Adap

Base model

Qwen/Qwen3-4B-Base
Finetuned
Qwen/Qwen3-4B
Adapter
(113)
this model

Dataset used to train laurent-maille/Valuoty-industry-plc-4B-V0.2Adap

Evaluation results