TorchCompileControlNet

TorchCompileControlNet 노드는 내부적으로 torch.compile을 사용합니다.

TorchCompileControlNet은 ControlNet 모델의 추론 성능을 최적화하도록 설계된 유틸리티 노드입니다. PyTorch의 JIT(Just-in-Time) 컴파일 기술을 활용하여 ControlNet 모델을 더 효율적인 실행 형식으로 변환하며, 이미지 생성 워크플로의 처리 속도를 향상시키는 것을 목표로 합니다.

노드 기능

이 노드는 내부적으로 torch.compile 함수를 사용하여 입력된 ControlNet 모델을 컴파일합니다. 사용자가 다양한 컴파일 백엔드(예: inductor 또는 cudagraphs)와 최적화 모드(예: max-autotune)를 선택할 수 있으며, 더 적극적인 최적화를 위한 전체 그래프 모드 활성화를 지원합니다. 컴파일 과정은 일회성이며, 노드 내부에서 컴파일 상태를 추적하여 중복 연산을 방지합니다.

노드 파라미터 설명 - TorchCompileControlNet

연결 입력 (Inputs)

파라미터 이름	데이터 타입	필수	기본값	범위/옵션	설명
`controlnet`	CONTROL_NET	예	-	-	컴파일 및 최적화할 ControlNet 모델을 입력합니다.

컨트롤 파라미터 (Parameters)

파라미터 이름	데이터 타입	필수	기본값	범위/옵션	설명
`backend`	COMBO	예	inductor	"inductor", "cudagraphs"	PyTorch 컴파일에 사용할 백엔드 엔진을 선택합니다. 옵션은 `inductor` 또는 `cudagraphs`입니다.
`mode`	COMBO	예	default	"default", "max-autotune", "max-autotune-no-cud...	컴파일 최적화 모드를 선택합니다. 옵션은 `default`, `max-autotune`, `max-autotune-no-cudagraphs` 또는 `reduce-overhead`입니다.
`fullgraph`	BOOLEAN	예	False	-	전체 그래프 모드를 활성화합니다.

출력 (Outputs)

파라미터 이름	데이터 타입	설명
CONTROL_NET	CONTROL_NET	컴파일 및 최적화된 ControlNet 모델을 출력합니다.

사용 시나리오

빠른 반복 또는 배치 이미지 생성이 필요한 워크플로에서, 이 노드를 ControlNet 로더 뒤, 샘플러 앞에 연결하여 사용할 수 있습니다. ControlNet을 컴파일함으로써 생성 품질을 유지하면서 단일 또는 다중 샘플링에 소요되는 시간을 현저히 줄일 수 있으며, 특히 생성 속도가 중요한 실시간 애플리케이션이나 대규모 콘텐츠 제작 시나리오에 적합합니다.

주의사항

이 노드는 실험적 기능으로 표시되어 있습니다. 컴파일 효과와 안정성은 PyTorch 버전, 하드웨어 드라이버 및 특정 모델에 따라 다를 수 있습니다. 컴파일이 실패하면 노드는 런타임 오류를 발생시키며, 환경 호환성을 확인하거나 다른 컴파일 파라미터 조합을 시도해야 합니다.