Image-Text-to-Text
Transformers
TensorBoard
Safetensors
gemma4
Generated from Trainer
conversational
MyModelCleanSFT / README.md
AlexHung29629's picture
Upload folder using huggingface_hub
cc621e8 verified
|
Raw History Blame Contribute Delete
5.72 kB
---
library_name: transformers
base_model: google/gemma-4-26B-A4B-it
tags:
- generated_from_trainer
datasets:
- data/ultradata-sft-clean.jsonl
- data/ultradata-sft-agent-clean.jsonl
- data/tw-sft-clean.jsonl
model-index:
- name: outputs/gemma4-26b-fft-clean-noeaft
results: []
---
<!-- 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.19.0.dev0`
```yaml
# 4 節點 × 8 張 H100,全參數 FSDP2。clean 版資料,純 SFT。
# 從 train-clean.yml 改來,只拿掉 EAFT。由 train-clean-noeaft.sbatch 送出。
# 不要改 train-clean.yml,那份是 EAFT(alpha 0.333)的對照。
# 不要在登入節點跑 preprocess 或 train。
# global batch = 32 * micro_batch_size * gradient_accumulation_steps。
# output 跟 train-clean.yml 分開,避免蓋到 gemma4-26b-fft-clean。
# 資料、長度、packing 跟 clean 65536 一樣,preprocess 可沿用那份 cache。
base_model: google/gemma-4-26B-A4B-it
# Gemma 4 權重是 Gemma4ForConditionalGeneration,不要強制 AutoModelForCausalLM。
# 這次是文字 SFT,凍結 vision / audio tower。
freeze_mm_modules: true
plugins:
- axolotl.integrations.liger.LigerPlugin
- axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
liger_rope: true
liger_rms_norm: true
liger_glu_activation: true
liger_layer_norm: true
# multimodal Gemma 4 的 liger 沒有 fused linear CE。
liger_fused_linear_cross_entropy: false
# 純交叉熵。CCE 不物化整段 logits,並套用 final_logit_softcapping=30。
# 65536 × vocab 262144 的 bf16 logits 約 32GiB,跟 EAFT 那次的佔用加起來放不進 H100。
# 容器的 fused EAFT 沒有 softcap。這次和 train-clean.yml 的差別是 entropy 加權,以及 softcap。
# 不要跟 use_eaft 同時開。
cut_cross_entropy: true
use_eaft: false
# 全參數。不要加 adapter、4bit、quantize_moe_experts。
load_in_8bit: false
load_in_4bit: false
datasets:
- path: data/ultradata-sft-clean.jsonl
type: input_output
- path: data/ultradata-sft-agent-clean.jsonl
type: input_output
- path: data/tw-sft-clean.jsonl
type: input_output
dataset_prepared_path: ./prepared/gemma4-26b-fft-clean-65536
output_dir: ./outputs/gemma4-26b-fft-clean-noeaft
dataset_num_proc: 64
train_on_inputs: false
val_set_size: 0
seed: 42
# 跟 train-clean.yml 相同:65536、gradient_accumulation_steps 1。
# 超過這個長度的樣本,Axolotl preprocess 會丟掉。
sequence_len: 65536
sample_packing: true
eval_sample_packing: false
pad_to_sequence_len: true
wandb_project:
wandb_entity:
wandb_watch:
wandb_name:
wandb_log_model:
use_tensorboard: true
# 32 張 H100。global batch = 32 * 1 * 1 * 65536 = 2,097,152 tokens。
gradient_accumulation_steps: 1
micro_batch_size: 1
num_epochs: 2
optimizer: adamw_torch_fused
learning_rate: 5.0e-6
lr_scheduler: cosine
warmup_ratio: 0.03
weight_decay: 0.01
max_grad_norm: 1.0
bf16: auto
tf32: true
gradient_checkpointing: true
gradient_checkpointing_kwargs:
use_reentrant: false
# 183274 打開後,backward 仍是已佔 72.23GiB 再要 5.50GiB。這段 workspace 沒被 offload 移走。
activation_offloading: false
# FSDP2 把參數、梯度、optimizer state 切到 32 張卡。
# 必須包文字 decoder。不指定時 FSDP 會去找不存在的 Gemma4AudioLayer。
# 存檔用 FULL_STATE_DICT,集中成一份權重。不寫 optimizer state,不能從 optimizer 續訓。
fsdp_version: 2
fsdp_config:
offload_params: false
cpu_ram_efficient_loading: true
auto_wrap_policy: TRANSFORMER_BASED_WRAP
transformer_layer_cls_to_wrap: Gemma4TextDecoderLayer
state_dict_type: FULL_STATE_DICT
reshard_after_forward: true
# flex attention。全域層 head_dim=512,FA2 單獨跑不了。
# 參數照 MEGA/train.yml。H100 預設 flex 會 SRAM OOM。
attn_implementation: flex_attention
flex_attn_kernel_options:
num_stages: 1
bwd_num_stages: 1
logging_steps: 1
saves_per_epoch: 1
save_total_limit: 2
# 只存模型參數。optimizer、scheduler、rng 都不寫進 checkpoint。
save_only_model: true
special_tokens:
bos_token: "<bos>"
eos_token: "<eos>"
pad_token: "<pad>"
unk_token: "<unk>"
# preprocess --debug 若把要學的 <turn|> 標成 -100,拿掉這段。
eot_tokens:
- "<turn|>"
```
</details><br>
# outputs/gemma4-26b-fft-clean-noeaft
This model is a fine-tuned version of [google/gemma-4-26B-A4B-it](https://huggingface.co/google/gemma-4-26B-A4B-it) on the data/ultradata-sft-clean.jsonl, the data/ultradata-sft-agent-clean.jsonl and the data/tw-sft-clean.jsonl datasets.
## 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: 5e-06
- train_batch_size: 1
- eval_batch_size: 1
- seed: 42
- distributed_type: multi-GPU
- num_devices: 32
- total_train_batch_size: 32
- total_eval_batch_size: 32
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED 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: 21
- training_steps: 712
### Training results
### Framework versions
- Transformers 5.14.1
- Pytorch 2.11.0+cu130
- Datasets 4.8.4
- Tokenizers 0.22.2