ComfyUI Extension
comfyui_LLM_schools
ComfyUI node library for fine-tuning LLMs
heshengtao/comfyui_LLM_schools
Nodes9
On cloudLocal install
Category大模型学校(llm_schools)/模型训练(Model Training), 大模型学校(llm_schools)/数据预处理(data preprocessing)
Stars7
Updated2 years ago
Nodes (9)
Causal Language Model Trainer
大模型学校(llm_schools)/模型训练(Model Training)
download/load the HF dataset
大模型学校(llm_schools)/数据预处理(data preprocessing)
get HF dataset repo_id
大模型学校(llm_schools)/数据预处理(data preprocessing)
IA3 Arguments
大模型学校(llm_schools)/模型训练(Model Training)
LLM Arguments
大模型学校(llm_schools)/模型训练(Model Training)
LoRA or Adapter Arguments
大模型学校(llm_schools)/模型训练(Model Training)
P or Prompt Arguments
大模型学校(llm_schools)/模型训练(Model Training)
Prefix Arguments
大模型学校(llm_schools)/模型训练(Model Training)
split HF dataset
大模型学校(llm_schools)/数据预处理(data preprocessing)
Readme
comfyui_LLM_schools
<div align="center"> <a href="./README.md"><img src="https://img.shields.io/badge/English-d9d9d9"></a> <a href="./README_ZH.md"><img src="https://img.shields.io/badge/简体中文-d9d9d9"></a> </div>- Currently supports full fine-tuning of AutoModelForCausalLM classes in the transformer library, refer to workflow The_first_training