ComfyUI Node
Causal Language Model Trainer
A ComfyUI node in 大模型学校（llm_schools）/模型训练（Model Training） with 7 inputs and 1 output.
Causal Language Model Trainer
- args
- peft_args
- log
◄split_datapaths►
◄model_name_or_pathgpt2►
◄deviceauto►
◄dtypefloat32►
◄fine_tuning_methodfull_fine_tuning►
Category大模型学校（llm_schools）/模型训练（Model Training）
Inputs (7)
| Name | Type | Default | Description |
|---|---|---|---|
| split_datapaths | STRING | — | |
| model_name_or_path | STRING | gpt2 | — |
| device | COMBO | auto | 4 options: auto, cuda, cpu, mps |
| dtype | COMBO | float32 | 5 options: float32, float16, bfloat16, int8, int4 |
| args | ARGS | [object Object] | — |
| fine_tuning_method | COMBO | full_fine_tuning | 7 options: full_fine_tuning, lora, adapter, p-tuning, prompt_tuning, prefix-tuning, +1 |
| peft_argsopt | ARGS | [object Object] | — |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| log | STRING | — |