ComfyUI Node: Causal Language Model Trainer

Authored by heshengtao

Created

Updated

8 stars

Category

大模型学校(llm_schools)/模型训练(Model Training)

Inputs

split_datapaths STRING
model_name_or_path STRING
device
  • auto
  • cuda
  • cpu
  • mps
dtype
  • float32
  • float16
  • bfloat16
  • int8
  • int4
args ARGS
fine_tuning_method
  • full_fine_tuning
  • lora
  • adapter
  • p-tuning
  • prompt_tuning
  • prefix-tuning
  • IA3
peft_args ARGS

Outputs

STRING

Extension: comfyui_LLM_schools

ComfyUI node library for fine-tuning LLMs

Authored by heshengtao

Run ComfyUI workflows in the Cloud!

No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues

Learn more