ComfyUI Node
P or Prompt Arguments
A ComfyUI node in 大模型学校（llm_schools）/模型训练（Model Training） with 9 inputs and 1 output.
P or Prompt Arguments
- peft_args
◄num_virtual_tokens20►
◄token_dim768►
◄num_transformer_submodules1►
◄num_attention_heads12►
◄num_layers12►
◄prompt_tuning_initRANDOM►
◄prompt_tuning_init_text►
◄tokenizer_name_or_pathbert-base-uncased►
◄is_enabletrue►
Category大模型学校（llm_schools）/模型训练（Model Training）
Inputs (9)
| Name | Type | Default | Description |
|---|---|---|---|
| num_virtual_tokens | INT | 201–100 | — |
| token_dim | INT | 7681–4096 | — |
| num_transformer_submodules | INT | 11–12 | — |
| num_attention_heads | INT | 121–16 | — |
| num_layers | INT | 121–24 | — |
| prompt_tuning_init | COMBO | RANDOM | 2 options: RANDOM, TEXT |
| prompt_tuning_init_text | STRING | — | |
| tokenizer_name_or_path | STRING | bert-base-uncased | — |
| is_enable | BOOLEAN | true | — |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| peft_args | ARGS | — |