pytorch를 사용하여 스테이블 디퓨전 파인튜닝하는 뼈대 코드 공유합니다.
허깅페이스 디퓨저 라이브러리 도움말이나 블로그에 보면 디퓨저에서 제공하는 학습 스크립트를 사용하는 예제만 나와있습니다.
Keras 에는 Fine-tuning Stable Diffusion 처럼 밑바닥부터 트레이닝 루프를 다 코딩하는 예제가 있는데 파이토치 사용해서는 이렇게 하는 예제를 찾기가 쉽지 않아 직접 만들어봤습니다.
diffusers에서 제공하는 스크립트 파일 사용하지 않고 학습 루프를 다 코딩하는 방식의 제일 접근하기 쉬운 바닐라 버전이라 입문하시는 분들 이해하기 쉬우실 듯 합니다. 그냥 죽 눈으로 따라만 보셔도 금방 이해가 되실것으로 생각합니다.
pytorch-stable-diffusion-fine-tuning
그럼 누군가에게 유용한 정보가 되길 바라겠습니다.
