NVIDIA 초해상화 모델 PID 1.5 공개: 몇 초 만에 2K 변환, 디테일 강화

PID 1.5는 NVIDIA가 공개한 오픈소스 확산 디코더로, 기존 VAE를 대체해 잠재 공간의 데이터를 단 한 번의 과정으로 초고해상도 픽셀 이미지로 변환합니다.

일반적인 AI 이미지 생성 환경에서는 모델이 저해상도 이미지를 만든 뒤 VAE로 복원하고, 별도의 업스케일러를 돌리는 번거로운 방식을 씁니다. PID는 디코딩과 업스케일링을 조건부 픽셀 확산 모델 하나로 통합해 이 단계를 하나로 합쳤습니다.

속도 역시 인상적입니다. 일반 소비자용 그래픽카드에서도 512×512 크기의 잠재 공간 이미지를 단 몇 초 만에 2K 또는 4K 해상도로 변환해 냅니다.

1.5 버전에서 개선된 점

1.0 버전과 비교했을 때 가장 돋보이는 부분은 세부 디테일의 선명도와 정확해진 색감입니다.

이전 버전에서 종종 발생하던 모서리 부분의 격자 노이즈 현상이 완전히 사라졌고, 색상 왜곡도 깔끔하게 잡혔습니다.

또한 1.5 버전에는 애니메이션 데이터와 원거리의 작은 얼굴 데이터가 대폭 추가 학습되었습니다. 배경에 작게 들어간 인물의 이목구비나 섬세한 머리칼 표현이 뭉개지지 않고 또렷하게 살아납니다.

지원 범위도 넓어졌습니다. 기존 FLUX와 Z-Image는 물론 FLUX.2와 Qwen-Image 등 최신 오픈소스 모델과 결합할 수 있으며, 1:1, 16:9, 9:16 등 다양한 화면비를 자연스럽게 지원합니다.

설치 및 실행 방법

NVIDIA는 관련 코드와 가중치, 학습 스크립트까지 GitHub와 Hugging Face에 모두 공개했습니다.

파이썬 환경에 익숙하다면 PyTorch와 CUDA 환경에서 uv나 pip 명령어로 손쉽게 세팅할 수 있습니다. 텍스트 프롬프트 기반 생성과 기존 이미지를 변환하는 공식 스크립트가 기본 제공됩니다.

일반 창작자들을 위해 ComfyUI 노드도 이미 지원합니다. 기존 워크플로에서 VAE 디코더 자리에 PID를 연결하기만 하면 곧바로 고해상도 이미지를 뽑아낼 수 있습니다.

다만 개발진의 설명에 따르면 오직 2048 픽셀 크기로만 뽑을 때는 기존 2K 전용 체크포인트가 미세하게 더 날카로울 수 있습니다. 1.5 버전은 2K에서 4K까지 유연하게 해상도를 확장하며 아티팩트를 줄이는 데 최적화되었습니다.

왜 중요한가

FLUX나 Qwen-Image를 PC에서 직접 구동하는 크리에이터라면 당장 워크플로에 적용해 볼 만합니다. 무거운 업스케일러 노드를 따로 붙이지 않아도 되어 VRAM을 아끼고 렌더링 시간을 크게 단축할 수 있습니다.

디코더 자체에 초해상화 능력을 내장해 파이프라인을 단순화한 NVIDIA의 설계는 매우 실용적입니다. 복잡해지기 쉬운 로컬 AI 생성 환경을 한 단계 간결하게 만들어 준 도구입니다.