ValueError: Cannot load <class 'diffusers.models.controlnet.ControlNetModel'> from / because the following keys are missing: Please make sure to pass `low_cpu_mem_usage=False` and `device_map=None` if you want to randomly initialize those weights or else

[전체 에러문구]

ValueError: Cannot load <class 'diffusers.models.controlnet.ControlNetModel'> from / because the following keys are missing:
 Please make sure to pass `low_cpu_mem_usage=False` and `device_map=None` if you want to randomly initialize those weights or else make sure your checkpoint file is correct.

 

아래 코드로 controlnet 모델을 로드 할  때 발생하는 에러입니다.

controlnet = ControlNetModel.from_pretrained('controlnet',
                                            torch_dtype = torch.float16,
                                            )

 

low_cpu_mem_usage, device_map 두 가지 parameter를 추가해 해결할 수 있습니다. 

controlnet = ControlNetModel.from_pretrained('controlnet',
                                            torch_dtype = torch.float16,
                                            low_cpu_mem_usage=False,
                                            device_map=None
                                            )