그림 AI를 만드는 방법

그림 AI를 만들기 위해서 필요한 데이터는 이미지와 텍스트가 1개의 쌍을 이루는 데이터셋을 사용하는데

제대로된 그림 AI 모델을 만들기 위한 이미지가 엄청 많이 필요하다. 

SD1.4을 학습하는데 사용한 데이터셋은 LAION-5B 라고 해서 58억 5천만 개의 이미지가 사용되었으니까 사실상 인터넷에서 돌아다니는 이미지를 전부 학습용으로 집어 넣는다고 이해하면 편하다. 만약 억 단위 미만으로 만들게 된다면 AI의 성능이 처참하게 되어버린다는 결론은 여러번 증명된 바 있다.

그리고 나서 미리 만들어놓은 VAE랑 텍스크 인코더로 이미지와 텍스트를 인코딩한 다음 적당히 학습 돌리면 그림 AI가 완성된다. 

이때 필요한 GPU 자원은 보통 수십억 규모라서 아무리 못해도 대학원 수준의 기술과 자본이 필요하기에 포기하자.