728x90
반응형
- 전체 에러문구
Missing key(s) in state_dict: "clip_model.vision_tower.vision_model.embeddings.position_ids".
pretrain 모델을 load_state_dict를 사용하여 업로드 할 때 모델의 구조가 맞지 않아 발생하는 에러입니다.
- 해결방법
load_state_dic 함수의 파라미터로 strict=False 를 추가해주면 됩니다.
model.load_state_dict(ckpt, strict=False)
이 파라미터를 추가해주면 모델을 불러올 떄 불러올 수 있는 값들만 유동적으로 불러올 수 있습니다.
728x90
반응형