스테이블 디퓨전에서 사용할 수 있는 모델(Checkpoint) 파일은 두 가지가 있습니다. Safetensors와 Ckpt 파일인데요, 오늘은 이 두 파일의 장단점과 차이점에 대해서 알아보도록 하겠습니다.
두 파일 모두 tensorflow라는 딥러닝 모델을 저장하는 방법 중 하나이기는 하나 차이점이 몇 가지 있습니다.
Safetensors 파일이란?
Ai 모델을 만들기 위해서는 checkpoint 파일인 모델 파일이 필요합니다.
이런 모델 파일을 배포하고 다른 사람들과 공유하여 사용하기 위한 파일 형태가 safetensors입니다. HuggingFace가 주도적으로 개발하고 있습니다.
기존에는 ckpt 파일을 사용하였지만 여러가지 개선을 통해서 safetensors를 최근에는 많이 사용하고 있는 상태입니다.
Safetensors vs ckpt 차이점
- ckpt 파일은 모델의 구조나 코드를 저장하지 않아 별도의 파이썬 파일이 필요합니다.
- 파이썬 이외에 다른 언어나 프레임워크와 호환되지 않습니다.
- 여러 개의 파일로 구성되어 있어 관리가 불편하고 용량이 큽니다.
가장 결정적인 차이점은 속도에 있습니다.
Safetensors가 ckpt에 비해 그림 생성 시간이 반으로 줄어들게 됩니다.
ckpt는 모델을 압축해 둔 상태라 풀면서 생기는 보안적 이슈가 있을 수 있는데 Safetensors는 이를 해결했을 뿐만 아니라 속도와 램 사용량이 더 좋아졌습니다. 스테이블 디퓨전을 통해서 두 파일은 병합이 가능합니다.
기회가 되면 ckpt 파일을 safetensors로 바꾸는 방법에 대해서 알아보도록 하겠습니다.
