ComfyUI Extension: ComfyUI-Extract_Flux_Lora

Authored by judian17

Created

Updated

7 stars

Extract LoRA from the original Fine-Tuned model.

Custom Nodes (0)

    README

    ComfyUI-Extract_Flux_Lora

    Extract LoRA from the original Fine-Tuned model.从微调模型中提取lora。

    此节点来自ComfyUI-FluxTrainer,用来从各种微调模型中提取lora。根据此issue修复了一个bug,原节点有些微调模型无法用来转换。注意,如果你已经安装了ComfyUI-FluxTrainer,那么为避免冲突你可以将本项目的flux_extract_lora.py代替ComfyUI-FluxTrainer里的同名文件。之所以提取lora,是因为目前nunchaku只能使用原版flux模型的svdq模型,无法使用其他微调过的模型。用此节点提取的lora可以与svdq模型一起使用,拥有svdquant的速度与显存占用,且效果与原微调模型比较接近,是nunchaku放出模型量化的节点前的临时替代方法。当然如果不在意差异,提取lora也可以代替原版微调模型,节省空间。提取的lora与原版模型对比结果如下图,对比用的lora的rank为128,适当提高强度能使得lora效果更接近原微调模型,详细内容可参考此文章reddit

    图片11 使用方法如下: image