Model Save KJ (ModelSaveKJ) - ComfyUI-KJNodes

기술 구현 관점에서 이 노드는 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)

파라미터 이름	데이터 타입	필수	기본값	범위/옵션	설명
`model`	MODEL	예	-	-	저장할 모델 객체를 지정합니다. 이는 노드 작업의 핵심 데이터 소스입니다.

컨트롤 파라미터 (Parameters)

파라미터 이름	데이터 타입	필수	기본값	범위/옵션	설명
`filename_prefix`	STRING	예	diffusion_models/ComfyUI	-	저장 파일의 경로 접두사와 기본 파일 이름을 정의합니다. 기본적으로 `output` 디렉토리 내의 `diffusion_models/ComfyUI` 경로에 저장되며, 파일 이름에는 자동으로 일련번호가 추가됩니다.
`model_key_prefix`	STRING	예	model.diffusion_model.	-	저장된 모델 파일에서 원본 키 이름 접두사를 대체하는 데 사용할 새로운 문자열을 지정합니다. 기본값은 `model.diffusion_model.`이며, 이 값을 수정하여 다른 프레임워크의 로딩 요구사항에 맞출 수 있습니다.

사용 시나리오

전형적인 사용 시나리오는 모델 병합 또는 파인튜닝 워크플로 이후입니다. 예를 들어, 여러 LoRA 모델을 사용하여 베이스 모델을 융합 조정한 후, 최종 생성된 모델을 이 노드를 통해 저장할 수 있습니다. 이렇게 하면 복잡한 전체 병합 과정을 반복하지 않고도 다른 프로젝트나 다른 ComfyUI 워크플로에서 이 사용자 정의 모델을 직접 로드할 수 있습니다.

Model Save KJ 노드 소스 코드 링크

Model Save KJ 노드는 ComfyUI-KJNodes 노드 패키지에서 제공됩니다.

Model Save KJ(ModelSaveKJ)

Model Save KJ