Diffusion Model Loader KJ (DiffusionModelLoaderKJ) - ComfyUI-KJNodes

Diffusion Model Loader KJ 노드는 모델을 로드할 때 가중치 데이터 유형과 계산 데이터 유형에 대한 세밀한 제어를 제공하며, fp8부터 fp32까지 다양한 정밀도 형식을 지원합니다.

Diffusion Model Loader KJ는 Stable Diffusion과 같은 확산 모델을 로드하고 구성하는 데 사용되는 노드입니다. 핵심 기능은 지정된 경로에서 모델 파일을 로드하고, 사용자가 런타임 시 모델의 수치 정밀도와 계산 동작을 조정하여 성능이나 호환성을 최적화할 수 있도록 합니다.

노드 기능

이 노드는 모델을 로드할 때 가중치 데이터 유형과 계산 데이터 유형에 대한 세밀한 제어를 제공하며, fp8부터 fp32까지 다양한 정밀도 형식을 지원합니다. 주요 기술 특징으로는 특정 GPU 가속 라이브러리를 활용하기 위한 선택적 CublasLinear 패치, SageAttention 어텐션 메커니즘 통합 지원, 그리고 추가 상태 사전을 통해 모델 기능을 병합하거나 강화하는 기능(예: 비디오 모델에 VACE 모듈 추가)이 포함됩니다.

노드 매개변수 설명 - Diffusion Model Loader KJ

연결 입력 (Inputs)

매개변수 이름	데이터 유형	필수	기본값	범위/옵션	설명
`model_name`	COMBO	예	-	-	로드할 확산 모델 파일을 선택합니다. 이 매개변수는 ComfyUI가 관리하는 모델 디렉토리에서 사용 가능한 모든 모델을 나열합니다.

컨트롤 매개변수 (Parameters)

매개변수 이름	데이터 유형	필수	기본값	범위/옵션	설명
`weight_dtype`	COMBO	예	default	default, fp8_e4m3fn, fp8_e4m3fn_fast, fp8_e5m2, fp16, bf16, fp32	메모리 사용량과 계산 정밀도 간의 균형을 맞춥니다.
`compute_dtype`	COMBO	예	default	default, fp16, bf16, fp32	계산 속도와 수치적 안정성에 영향을 줍니다.
`patch_cublaslinear`	BOOLEAN	예	False	-	cublas_ops 인수의 활성화 또는 비활성화
`sage_attention`	COMBO	예	False	false, true	comfy attention을 sageattn을 사용하도록 패치합니다.
`enable_fp16_accumulation`	BOOLEAN	예	False	-	torch.backends.cuda.matmul.allow_fp16_accumulation을 활성화합니다. pytorch 2.7.0 nightly 버전이 필요합니다.
`extra_state_dict`	STRING	아니오	-	-	로드할 추가 상태 사전의 전체 경로입니다. 이는 메인 상태 사전과 병합됩니다. 예를 들어 WanVideoModel에 VACE 모듈을 추가하는 데 유용합니다. DiffusionModelSelector를 사용하여 경로를 쉽게 얻을 수 있습니다.

출력 (Outputs)

매개변수 이름	데이터 유형	설명
MODEL	MODEL	로드되고 구성된 모델을 출력하여 후속 노드에서 사용할 수 있도록 합니다.

사용 시나리오

기본 모델을 로드하고 특정 최적화나 기능 확장을 적용해야 하는 워크플로에서 이 노드는 매우 유용합니다. 예를 들어, 이 노드를 사용하여 WanVideoModel을 먼저 로드한 다음, extra_state_dict 매개변수를 통해 VACE 모듈의 상태 사전을 병합하여 기능이 향상된 비디오 생성 모델을 구축할 수 있습니다.

주의사항

이 노드는 실험적 카테고리에 속하며, 일부 기능(예: enable_fp16_accumulation)은 특정 버전의 PyTorch(예: 2.7.0 nightly)가 필요합니다. 고급 데이터 유형(예: fp8)이나 패치 기능을 사용하기 전에 하드웨어 및 소프트웨어 환경이 지원하는지 확인하십시오.

Diffusion Model Loader KJ 노드 소스 코드 링크

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

Diffusion Model Loader KJ(DiffusionModelLoaderKJ)