Model Save KJ(ModelSaveKJ)

Узел Model Save KJ, с технической точки зрения, принимает входные данные типа MODEL и получает словарь параметров модели через метод state_dict_for_saving.

Model Save KJ — это узел, предназначенный для сохранения весов диффузионной модели после обработки или слияния. Его основная функция — сохранять модель из рабочего процесса на диск в формате .safetensors, позволяя пользователю настраивать путь сохранения и префикс ключей внутри модели. Узел не выводит никаких данных, а только выполняет операцию сохранения.

Функциональность узла

С технической точки зрения, этот узел принимает входные данные типа MODEL и получает словарь параметров модели через метод state_dict_for_saving. Его ключевой механизм заключается в том, что он позволяет пользователю указать model_key_prefix для замены префикса по умолчанию (model.diffusion_model.) в ключах словаря параметров модели, после чего обработанный словарь сохраняется в файл .safetensors. Это обеспечивает гибкость для последующей загрузки в другие фреймворки или инструменты, поддерживающие различные соглашения об именовании ключей.

Описание параметров узла - Model Save KJ

Входные соединения (Inputs)

Параметры управления (Parameters)

Сценарии использования

Типичный сценарий использования — после рабочего процесса слияния или тонкой настройки модели. Например, после объединения нескольких моделей LoRA с базовой моделью для адаптации, итоговую модель можно сохранить с помощью этого узла. Это позволяет загрузить эту пользовательскую модель в других проектах или рабочих процессах ComfyUI без необходимости повторения всего сложного процесса слияния.

Ссылка на исходный код узла Model Save KJ

Узел Model Save KJ взят из пакета узлов ComfyUI-KJNodes.